Home
last modified time | relevance | path

Searched refs:CompilationResult (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.h77 const XlaCompiler::CompilationResult** out_compilation_result,
88 const XlaCompiler::CompilationResult** out_compilation_result,
131 XlaCompiler::CompilationResult*)>& compile_fn,
133 const XlaCompiler::CompilationResult** out_compilation_result,
139 const XlaCompiler::CompilationResult& result,
159 XlaCompiler::CompilationResult compilation_result TF_GUARDED_BY(mu);
Dxla_compile_on_demand_op.h47 const XlaCompiler::CompilationResult** result,
53 const XlaCompiler::CompilationResult* result,
Dxla_launch_util.h87 const XlaCompiler::CompilationResult& compilation_result,
158 const XlaCompiler::CompilationResult* compilation_result,
175 const XlaCompiler::CompilationResult* compilation_result,
Dxla_compilation_cache.cc159 const XlaCompiler::CompilationResult& result, in BuildExecutable()
190 const XlaCompiler::CompilationResult** out_compilation_result, in Compile()
197 XlaCompiler::CompilationResult* result) { in Compile()
266 const XlaCompiler::CompilationResult** out_compilation_result, in CompileSingleOp()
277 XlaCompiler::CompilationResult* result) { in CompileSingleOp()
333 XlaCompiler::CompilationResult*)>& compile_fn, in CompileImpl()
335 const XlaCompiler::CompilationResult** out_compilation_result, in CompileImpl()
Dxla_compile_on_demand_op.cc46 const XlaCompiler::CompilationResult* result, in Run()
103 OpKernelContext* ctx, const XlaCompiler::CompilationResult** result, in Compile()
166 const XlaCompiler::CompilationResult* result; in Compute()
Dget_compiler_ir.cc41 const XlaCompiler::CompilationResult* compilation_result = nullptr; in GetLocalExecutable()
124 XlaCompiler::CompilationResult result; in GetCompilerIr()
Dxla_launch_util.cc222 const XlaCompiler::CompilationResult* compilation_result, in PopulateInputs()
345 const XlaCompiler::CompilationResult* compilation_result, int output_num) { in SetOutputForConstant()
400 const XlaCompiler::CompilationResult& compilation_result, in GatherVariableInfo()
422 const XlaCompiler::CompilationResult* compilation_result, in PopulateOutputs()
Dxla_compilation_cache_test.cc66 const XlaCompiler::CompilationResult* compilation_result; in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc55 XlaCompiler::CompilationResult* then_result, in PopulateTensorArrayGradients()
56 XlaCompiler::CompilationResult* else_result) { in PopulateTensorArrayGradients()
58 for (XlaCompiler::CompilationResult* result : {then_result, else_result}) { in PopulateTensorArrayGradients()
88 XlaOpKernelContext* ctx, const XlaCompiler::CompilationResult& then_result, in ValidateShapes()
89 const XlaCompiler::CompilationResult& else_result) { in ValidateShapes()
249 XlaCompiler::CompilationResult then_result; in Compile()
252 XlaCompiler::CompilationResult else_result; in Compile()
334 for (XlaCompiler::CompilationResult* result : {&then_result, &else_result}) { in Compile()
Dxla_select_and_scatter_op.cc72 XlaCompiler::CompilationResult select; in Compile()
93 XlaCompiler::CompilationResult scatter; in Compile()
Dwhile_op.cc173 const XlaCompiler::CompilationResult& body, bool has_token_input_output) { in VerifyBodyInputAndOutputShapeMatch()
198 XlaOpKernelContext* ctx, const XlaCompiler::CompilationResult& cond) { in BuildWrappedCond()
209 XlaOpKernelContext* ctx, const XlaCompiler::CompilationResult& body, in BuildWrappedBody()
341 XlaCompiler::CompilationResult body; in Compile()
433 XlaCompiler::CompilationResult cond; in Compile()
Dcase_op.cc162 std::vector<XlaCompiler::CompilationResult> branch_results(num_branches); in Compile()
170 for (XlaCompiler::CompilationResult& result : branch_results) { in Compile()
349 for (const XlaCompiler::CompilationResult& result : branch_results) { in Compile()
Dgather_scatter_ops.cc81 XlaCompiler::CompilationResult update_computation; in Compile()
Dreduce_window_op.cc86 XlaCompiler::CompilationResult reducer; in Compile()
Dsort_ops.cc91 XlaCompiler::CompilationResult comparator; in Compile()
Dxla_reduce_op.cc82 XlaCompiler::CompilationResult reducer; in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc185 XlaCompiler::CompilationResult result; in TEST_F()
216 XlaCompiler::CompilationResult result; in TEST_F()
267 XlaCompiler::CompilationResult result; in TEST_F()
317 XlaCompiler::CompilationResult result; in TEST_F()
363 XlaCompiler::CompilationResult result; in TEST_F()
413 XlaCompiler::CompilationResult result; in TEST_F()
456 XlaCompiler::CompilationResult result; in TEST_F()
478 XlaCompiler::CompilationResult result; in TEST_F()
527 XlaCompiler::CompilationResult result; in TEST_F()
558 XlaCompiler::CompilationResult result; in TEST_F()
[all …]
Dxla_compiler.h139 using CompilationResult = ::tensorflow::XlaCompilationResult; variable
216 CompilationResult* result);
224 CompilationResult* result);
348 CompilationResult, SignatureHash>
Dtf2xla.cc81 XlaCompiler::CompilationResult result; in ConvertGraphToXla()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.h125 const XlaCompiler::CompilationResult& compilation_result,
133 const XlaCompiler::CompilationResult& compilation_result,
138 const XlaCompiler::CompilationResult& compilation_result,
Dtpu_util.h47 const XlaCompiler::CompilationResult& compilation_result);
Dtpu_compile_op_support.cc201 const XlaCompiler::CompilationResult& compilation_result, in AddVariableUpdatesToCores()
271 const XlaCompiler::CompilationResult& compilation_result, in ComputeOutputShapesForEachCore()
309 const tensorflow::XlaCompiler::CompilationResult& compilation_result, in CreateHloModules()
Dtpu_util.cc55 const XlaCompiler::CompilationResult& compilation_result) { in BuildAotXlaComputationInstance()
Dtpu_compile_op_common.h182 XlaCompiler::CompilationResult* compilation_result);
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc79 const XlaCompiler::CompilationResult* compilation_result, in XlaExecutableClosure()
92 const XlaCompiler::CompilationResult* compilation_result() const { in compilation_result()
103 const XlaCompiler::CompilationResult* compilation_result_;
171 const XlaCompiler::CompilationResult** compilation_result, in CompileToLocalExecutable()
225 const XlaCompiler::CompilationResult* compilation_result; in Compute()
374 const XlaCompiler::CompilationResult* kernel; in Compute()

12