Home
last modified time | relevance | path

Searched defs:flr (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator.cc39 static bool IsCompilable(FunctionLibraryRuntime* flr, const NodeDef& ndef, in IsCompilable()
74 const FunctionLibraryRuntime& flr, in CanCreateKernel()
80 static Status CreateXlaKernel(FunctionLibraryRuntime* flr, in CreateXlaKernel()
173 FunctionLibraryRuntime* flr, in CreateKernel()
Dforce_xla_constants_on_host_pass.cc32 FunctionLibraryRuntime* flr = in Run() local
Dget_compiler_ir.cc56 FunctionLibraryRuntime* flr = pflr->GetFLR(dev->name()); in GetCompilerIr() local
Dextract_outside_compilation_pass.cc1809 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForFuncCallNode()
1895 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForIfNode()
2014 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForWhileNode()
2114 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2311 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction()
2503 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilation()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_threadpool_test.cc77 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()
Dprocess_function_library_runtime.cc156 FunctionLibraryRuntime* flr = nullptr; in GetRetTypes() local
177 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation() local
188 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext() local
733 FunctionLibraryRuntime* flr = GetFLR(options.target); in InstantiateMultiDevice() local
990 FunctionLibraryRuntime* flr = GetFLR(opts.target); in InstantiateMultiDevice() local
1170 FunctionLibraryRuntime* flr = GetFLR(target); in RunMultiDevice() local
1210 FunctionLibraryRuntime* flr = GetFLR(options.target); in Instantiate() local
1307 FunctionLibraryRuntime* flr = GetFLR(device); in ReleaseMultiDeviceHandle() local
1338 FunctionLibraryRuntime* flr = nullptr; in ReleaseHandle() local
1479 FunctionLibraryRuntime* flr = nullptr; in RunInternal() local
[all …]
Dfunction_test.cc177 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
215 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
221 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
228 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
238 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
265 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
297 Status InstantiateAndRunViaCallFrameInterface(FunctionLibraryRuntime* flr, in InstantiateAndRunViaCallFrameInterface()
334 std::unique_ptr<Graph> GetFuncBody(FunctionLibraryRuntime* flr, in GetFuncBody()
350 std::unique_ptr<Graph> GetGradBody(FunctionLibraryRuntime* flr, in GetGradBody()
Dprocess_function_library_runtime_test.cc343 FunctionLibraryRuntime* flr = in TEST_F() local
350 FunctionLibraryRuntime* flr = in TEST_F() local
1203 auto* flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F() local
1217 auto* flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F() local
Dfunction_utils.cc275 FunctionLibraryRuntime* flr, in InstantiateFunctionCall()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc142 FunctionLibraryRuntime* flr, bool* any_function_modified, in FunctionalizeControlFlowForNodeAssociatedFunctions()
207 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction()
289 FunctionLibraryRuntime* flr = in FunctionalizeControlFlow() local
339 FunctionLibraryRuntime* flr = in Run() local
Dtf2xla_util.h152 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h113 FunctionLibraryRuntime* flr, in KernelAndDevice()
194 FunctionLibraryRuntime* flr, in KernelAndDeviceOp()
261 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h80 FunctionLibraryRuntime* flr, in State()
98 FunctionLibraryRuntime* flr() { return flr_; } in flr() function
Ddataset_ops.cc136 FunctionLibraryRuntime* flr; in Compute() local
Dmulti_device_iterator_ops.cc60 FunctionLibraryRuntime* flr, in MultiDeviceIterator()
136 FunctionLibraryRuntime* const flr() { in flr() function in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
428 FunctionLibraryRuntime* flr; in Compute() local
Drewrite_utils.cc163 FunctionLibraryRuntime* flr = nullptr; in RewriteDataset() local
Diterator_ops.cc85 FunctionLibraryRuntime* flr) in IteratorResource()
474 FunctionLibraryRuntime* flr; in Compute() local
856 FunctionLibraryRuntime* flr; in TryInit() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op_test.cc139 auto flr = pflr_->GetFLR(device_->name()); in InitOpWithFunctionLibrary() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc458 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info) { in RunShapeInferenceOnComputation()
490 std::unique_ptr<Graph>* graph, FunctionLibraryRuntime* flr, in OptimizeGraph()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc48 FunctionLibraryRuntime* flr = ctx.func_lib(device); in CreateUncachedKernelAndDeviceOp() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1881 const FunctionBody& fbody, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
2266 Graph* g, Node* call_node, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsFromCallNode()
2352 Status LiftOutsideCompilationOnlyArgs(Graph* g, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgs()
2885 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld) { in ProcessHeadTailOutsideCompilation()
2970 FunctionLibraryRuntime* flr = in Run() local
Ddistributed_tpu_rewrite_pass.cc1662 const NameAttrList& function, FunctionLibraryRuntime* flr, in GetComputationForTPUReplicateOp()
1824 const Node& n, int num_cores_per_replica, FunctionLibraryRuntime* flr, in InferXlaShardingFromNeighbors()
1917 const Node* replicate_node, FunctionLibraryRuntime* flr, in AssignArgsAndRetvalsToCores()
3972 int num_replicas, const Node& replicate_node, FunctionLibraryRuntime* flr, in GetIOTypes()
4212 FunctionLibraryRuntime* flr, Node* host_compute_key_placeholder_node, in RewriteTPUReplicateNode()
4376 FunctionLibraryRuntime* flr) { in PerformHostTrainingLoopOptimization()
4441 FunctionLibraryRuntime* flr = in Run() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc268 FunctionLibraryRuntime* flr = pflr->GetFLR(cpu_device->name()); in RuntimeGraphOptimizer() local
/external/deqp-deps/glslang/glslang/MachineIndependent/
DConstant.cpp649 double flr = floor(unionArray[i].getDConst()); in fold() local
/external/angle/third_party/vulkan-deps/glslang/src/glslang/MachineIndependent/
DConstant.cpp648 double flr = floor(unionArray[i].getDConst()); in fold() local

12