experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF:XLA] Fix typo in function name.

Browse Source 
PiperOrigin-RevId: 217582093
This commit is contained in:
Peter Hawkins 2018-10-17 13:56:37 -07:00 committed by TensorFlower Gardener
parent 9fafb44f87
commit 610ebb2101
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2xla/tf2xla_util.cc
View File

@ -293,7 +293,7 @@ Status SetNodeShardingFromNeighbors(Node* n, bool out_edges) {
return Status::OK(); return Status::OK();
} }
void AddDtypeToKernalDefConstraint(absl::string_view name, DataType dtype, void AddDtypeToKernelDefConstraint(absl::string_view name, DataType dtype,
KernelDef* kdef) { KernelDef* kdef) {
for (KernelDef::AttrConstraint& constraint : *kdef->mutable_constraint()) { for (KernelDef::AttrConstraint& constraint : *kdef->mutable_constraint()) {
if (constraint.name() == name) { if (constraint.name() == name) {

2
tensorflow/compiler/tf2xla/tf2xla_util.h
View File

@ -55,7 +55,7 @@ string TensorIdToString(const tf2xla::TensorId& id);
Status SetNodeShardingFromNeighbors(Node* n, bool out_edges); Status SetNodeShardingFromNeighbors(Node* n, bool out_edges);
// Add an allowed data type to the AttrConstraint with the given name. // Add an allowed data type to the AttrConstraint with the given name.
void AddDtypeToKernalDefConstraint(absl::string_view name, DataType dtype, void AddDtypeToKernelDefConstraint(absl::string_view name, DataType dtype,
KernelDef* kdef); KernelDef* kdef);
// Returns the next random seed to use for seeding xla rng. // Returns the next random seed to use for seeding xla rng.

4
tensorflow/compiler/tf2xla/xla_cpu_backend.cc
View File

@ -21,10 +21,10 @@ namespace tensorflow {
bool CpuOpFilter(KernelDef* kdef) { bool CpuOpFilter(KernelDef* kdef) {
if (kdef->op() == "Const") { if (kdef->op() == "Const") {
AddDtypeToKernalDefConstraint("dtype", DT_STRING, kdef); AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef);
} }
if (kdef->op() == "Assert") { if (kdef->op() == "Assert") {
AddDtypeToKernalDefConstraint("T", DT_STRING, kdef); AddDtypeToKernelDefConstraint("T", DT_STRING, kdef);
} }
return true; return true;
} }

4
tensorflow/compiler/tf2xla/xla_gpu_backend.cc
View File

@ -21,10 +21,10 @@ namespace tensorflow {
bool GpuOpFilter(KernelDef* kdef) { bool GpuOpFilter(KernelDef* kdef) {
if (kdef->op() == "Const") { if (kdef->op() == "Const") {
AddDtypeToKernalDefConstraint("dtype", DT_STRING, kdef); AddDtypeToKernelDefConstraint("dtype", DT_STRING, kdef);
} }
if (kdef->op() == "Assert") { if (kdef->op() == "Assert") {
AddDtypeToKernalDefConstraint("T", DT_STRING, kdef); AddDtypeToKernelDefConstraint("T", DT_STRING, kdef);
} }
return true; return true;
} }