| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator.cc | 36 const FunctionLibraryRuntime& flr, in CanCreateKernel()
42 static Status CreateXlaKernel(FunctionLibraryRuntime* flr, in CreateXlaKernel()
85 FunctionLibraryRuntime* flr, in CreateKernel()
|
| D | force_xla_constants_on_host_pass.cc | 32 FunctionLibraryRuntime* flr = in Run() local
|
| D | xla_platform_info.cc | 46 Status BuildXlaCompilationCache(DeviceBase* device, FunctionLibraryRuntime* flr, in BuildXlaCompilationCache()
|
| D | get_compiler_ir.cc | 90 FunctionLibraryRuntime* flr = pflr->GetFLR(dev->name()); in GetCompilerIr() local
|
| D | extract_outside_compilation_pass.cc | 1812 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForFuncCallNode()
1898 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForIfNode()
2017 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForWhileNode()
2117 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2312 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction()
2504 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilation()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_threadpool_test.cc | 77 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
115 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
121 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
128 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
138 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
164 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
|
| D | process_function_library_runtime.cc | 159 FunctionLibraryRuntime* flr = nullptr; in GetRetTypes() local
180 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation() local
191 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext() local
822 FunctionLibraryRuntime* flr = GetFLR(options.target); in InstantiateMultiDevice() local
1133 FunctionLibraryRuntime* flr = GetFLR(opts.target); in InstantiateMultiDevice() local
1308 FunctionLibraryRuntime* flr = GetFLR(target); in RunMultiDeviceSync() local
1433 FunctionLibraryRuntime* flr = GetFLR(target); in RunMultiDeviceAsync() local
1474 FunctionLibraryRuntime* flr = GetFLR(options.target); in Instantiate() local
1571 FunctionLibraryRuntime* flr = GetFLR(device); in ReleaseMultiDeviceHandle() local
1602 FunctionLibraryRuntime* flr = nullptr; in ReleaseHandle() local
[all …]
|
| D | function_test.cc | 184 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
210 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
216 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
223 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
233 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
259 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
280 Status InstantiateAndRunViaCallFrameInterface(FunctionLibraryRuntime* flr, in InstantiateAndRunViaCallFrameInterface()
317 std::unique_ptr<Graph> GetFuncBody(FunctionLibraryRuntime* flr, in GetFuncBody()
333 std::unique_ptr<Graph> GetGradBody(FunctionLibraryRuntime* flr, in GetGradBody()
|
| D | process_function_library_runtime_test.cc | 337 FunctionLibraryRuntime* flr = in TEST_F() local
373 FunctionLibraryRuntime* flr = in TEST_F() local
1252 auto* flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F() local
1266 auto* flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F() local
|
| D | function_utils.cc | 275 FunctionLibraryRuntime* flr, in InstantiateFunctionCall()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 141 FunctionLibraryRuntime* flr, bool* any_function_modified, in FunctionalizeControlFlowForNodeAssociatedFunctions()
206 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction()
288 FunctionLibraryRuntime* flr = in FunctionalizeControlFlow() local
338 FunctionLibraryRuntime* flr = in Run() local
|
| D | tf2xla_util.h | 150 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 85 FunctionLibraryRuntime* flr, in State()
108 FunctionLibraryRuntime* flr() { return flr_; } in flr() function
|
| D | iterator_ops_test.cc | 47 FunctionLibraryRuntime* flr = nullptr; in GetIteratorResource() local
|
| D | dataset_ops.cc | 158 FunctionLibraryRuntime* flr; in Compute() local
|
| D | multi_device_iterator_ops.cc | 83 FunctionLibraryRuntime* flr, in MultiDeviceIterator()
156 FunctionLibraryRuntime* const flr() { in flr() function in tensorflow::data::__anon2efe57860111::MultiDeviceIterator
513 FunctionLibraryRuntime* flr; in Compute() local
|
| D | iterator_ops.cc | 95 FunctionLibraryRuntime* flr) in IteratorResource()
466 FunctionLibraryRuntime* flr; in Compute() local
786 FunctionLibraryRuntime* flr; in TryInit() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | serialization_utils.cc | 57 Status FromGraphDef(FunctionLibraryRuntime* flr, const GraphDef& graph_def, in FromGraphDef()
195 Status VariantTensorDataReader::ReadTensor(FunctionLibraryRuntime* flr, in ReadTensor()
207 Status VariantTensorDataReader::ReadTensor(FunctionLibraryRuntime* flr, in ReadTensor()
249 Status VariantTensorDataReader::ReadTensorInternal(FunctionLibraryRuntime* flr, in ReadTensorInternal()
270 Status VariantTensorDataReader::ReadDatasetInternal(FunctionLibraryRuntime* flr, in ReadDatasetInternal()
|
| D | captured_function.h | 143 FunctionLibraryRuntime* flr; member
|
| D | rewrite_utils.cc | 187 FunctionLibraryRuntime* flr = nullptr; in RewriteDataset() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 119 FunctionLibraryRuntime* flr, in KernelAndDevice()
202 FunctionLibraryRuntime* flr, in KernelAndDeviceOp()
271 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 166 std::unique_ptr<Graph>* graph, FunctionLibraryRuntime* flr, in OptimizeGraph()
341 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info) { in RunShapeInferenceOnComputation()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | op_kernel_runner.cc | 68 tensorflow::FunctionLibraryRuntime* flr, tensorflow::NodeDef ndef, in CreateOpKernel()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | op_kernel_runner.cc | 70 tensorflow::FunctionLibraryRuntime* flr, tensorflow::NodeDef ndef, in CreateOpKernel()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 178 auto flr = pflr_->GetFLR(device_->name()); in InitOpWithFunctionLibrary() local
|