| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 93 flib_.reset(new FunctionLibraryDefinition(graph_->flib_def())); in Init()
132 std::unique_ptr<FunctionLibraryDefinition> flib_; member in tensorflow::MlirGraphOptimizationPassTest
145 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
160 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
174 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
190 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
209 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
226 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 139 flib_(flib) {} in BatchResource()
151 flib_(flib) {} in BatchResource()
171 flib_->Run(opts, fhandle_, inputs, combined_outputs, in ProcessFuncBatchImpl()
184 FunctionLibraryRuntime* flib_; member in tensorflow::BatchResource
202 flib_ = c->function_library(); in BatchFunctionKernel()
292 handle, flib_, &new_resource)); in ComputeAsync()
301 max_enqueued_batches_, allowed_batch_sizes_, handle, flib_, in ComputeAsync()
323 if (!flib_) { in InstantiateFunction()
328 opts.target = flib_->device() == nullptr ? "" : flib_->device()->name(); in InstantiateFunction()
330 const ConfigProto* config = flib_->config_proto(); in InstantiateFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | hash_utils.cc | 153 : graph_(graph), root_(root), flib_(flib) {} in GraphHasher()
309 TF_RETURN_IF_ERROR(flib_->LookUp(node->op(), ®)); in HashNodeNonInput()
347 TF_RETURN_IF_ERROR(flib_->LookUp(this_node->op(), ®)); in CheckNodesEqualNonInput()
467 const FunctionDef* fdef = flib_->Find(name); in HashFunction()
472 FunctionDefToBodyHelper(*fdef, AttrSlice(&attrs), flib_, &fbody)); in HashFunction()
480 GraphHasher hasher(&graph_def, &ret_node->def(), flib_); in HashFunction()
521 const FunctionDef* this_fdef = flib_->Find(this_name); in CheckFunctionsEqualHelper()
522 const FunctionDef* that_fdef = that->flib_->Find(that_name); in CheckFunctionsEqualHelper()
527 *this_fdef, AttrSlice(&this_attrs), flib_, &this_fbody)); in CheckFunctionsEqualHelper()
531 *that_fdef, AttrSlice(&that_attrs), that->flib_, &that_fbody)); in CheckFunctionsEqualHelper()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 63 flib_(flib), in GraphCompiler()
83 FunctionLibraryRuntime* flib_; variable
|
| D | graph_compiler.cc | 139 Status s = flib_->CreateKernel(n->properties(), &op_kernel_raw); in Compile()
173 if (IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)) { in Compile()
228 TF_RET_CHECK(IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)); in CompileFunctionalNode()
239 TF_RETURN_IF_ERROR(GetFunctionNameAndAttr(*flib_, *n, &func)); in CompileFunctionalNode()
338 params->function_library = flib_; in PartiallySetupParams()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 349 FunctionLibraryRuntime* function_library() const { return flib_; } in function_library()
379 FunctionLibraryRuntime* flib_; variable
|
| D | op_kernel.cc | 238 flib_(flib), in OpKernelConstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 166 : flib_(flib) {} in NameUniquifier()
170 return !flib_.Contains(std::string(name)); in IsUnique()
178 const FunctionLibraryDefinition& flib_; member in tensorflow::__anon014ec6670111::NameUniquifier
|