Searched refs:kFuncAttr (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 43 ctx, ctx->GetAttr(FunctionLibraryDefinition::kFuncAttr, func_.get())); in PartitionedCallOp()
|
| D | function_ops.cc | 298 ctx->GetAttr(FunctionLibraryDefinition::kFuncAttr, &func_)); in RemoteCallOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.cc | 198 node.attrs().Find(FunctionLibraryDefinition::kFuncAttr, &attr_value)); in GetFunctionNameAndAttr()
|
| D | functionalize_while.cc | 267 AttrSlice(&node.attr()).Find(FunctionLibraryDefinition::kFuncAttr); in AddMissingFunctionDef()
|
| D | tf2xla_util.cc | 661 node->attrs(), FunctionLibraryDefinition::kFuncAttr, &func)); in RewriteAssociatedFunction()
|
| D | xla_compiler.cc | 966 node.attrs().Find(FunctionLibraryDefinition::kFuncAttr, &attr_value)); in GetPotentialFunctionName()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1339 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
1345 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
|
| D | function.h | 407 static constexpr const char* const kFuncAttr = "f"; variable
|
| D | function.cc | 1299 if (!GetNodeAttr(ndef, kFuncAttr, &forward_func_attrs).ok()) { in GetAttrImpl()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 59 static constexpr const char* const kFuncAttr = variable
60 FunctionLibraryDefinition::kFuncAttr;
704 const AttrValue* f = attrs.Find(kFuncAttr); in Instantiate()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1296 n->def(), FunctionLibraryDefinition::kFuncAttr, &original_func)); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
|