| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tf_xla_mlir_translate.cc | 69 const XlaCompilationResult& compilation_result, llvm::raw_ostream& output) { in PrintHloModuleText() argument
71 compilation_result.computation->GetProgramShape().ValueOrDie()); in PrintHloModuleText()
73 compilation_result.computation->proto(), module_config); in PrintHloModuleText()
84 if (!compilation_result.input_mapping.empty()) in PrintHloModuleText()
86 << absl::StrJoin(compilation_result.input_mapping, ", ") << "}\n"; in PrintHloModuleText()
88 for (const auto& xla_input_shape : compilation_result.xla_input_shapes) in PrintHloModuleText()
92 << compilation_result.xla_output_shape.ToString() << '\n'; in PrintHloModuleText()
94 for (const auto& xla_output_description : compilation_result.outputs) { in PrintHloModuleText()
106 for (const auto& resource_update : compilation_result.resource_updates) { in PrintHloModuleText()
242 llvm::StringRef device_type, XlaCompilationResult* compilation_result, in CompileMlirToXlaHloViaBuilder() argument
[all …]
|
| D | compile_mlir_util.h | 106 XlaCompilationResult* compilation_result);
116 XlaCompilationResult* compilation_result,
126 XlaCompilationResult* compilation_result,
139 XlaCompilationResult* compilation_result,
151 XlaCompilationResult* compilation_result,
|
| D | compile_mlir_util.cc | 450 XlaCompilationResult* compilation_result) { in PopulateResultIOInfo() argument
453 GetInputMappingForMlir(arg_shapes.size(), &compilation_result->input_mapping); in PopulateResultIOInfo()
458 &compilation_result->xla_input_shapes)); in PopulateResultIOInfo()
463 &compilation_result->xla_output_shape, &compilation_result->outputs, in PopulateResultIOInfo()
464 &compilation_result->resource_updates); in PopulateResultIOInfo()
472 XlaCompilationResult* compilation_result, in CompileMlirToXlaHlo() argument
479 compilation_result->computation = std::make_shared<xla::XlaComputation>(); in CompileMlirToXlaHlo()
481 module_op, device_type, compilation_result->computation.get(), in CompileMlirToXlaHlo()
487 shape_representation_fn, compilation_result); in CompileMlirToXlaHlo()
494 XlaCompilationResult* compilation_result, in CompileSerializedMlirToXlaHlo() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 222 const XlaCompiler::CompilationResult* compilation_result, in PopulateInputs() argument
227 arguments.reserve(compilation_result->xla_input_shapes.size()); in PopulateInputs()
231 for (int i = 0, end = compilation_result->xla_input_shapes.size(); i < end; in PopulateInputs()
233 int arg_num = compilation_result->input_mapping[i]; in PopulateInputs()
235 const xla::Shape& shape = compilation_result->xla_input_shapes[i]; in PopulateInputs()
242 absl::c_any_of(compilation_result->resource_updates, in PopulateInputs()
345 const XlaCompiler::CompilationResult* compilation_result, int output_num) { in SetOutputForConstant() argument
346 CHECK(compilation_result->outputs[output_num].is_constant); in SetOutputForConstant()
348 compilation_result->outputs[output_num].constant_value; in SetOutputForConstant()
400 const XlaCompiler::CompilationResult& compilation_result, in GatherVariableInfo() argument
[all …]
|
| D | xla_launch_util.h | 87 const XlaCompiler::CompilationResult& compilation_result,
158 const XlaCompiler::CompilationResult* compilation_result,
175 const XlaCompiler::CompilationResult* compilation_result,
|
| D | xla_compilation_cache_test.cc | 66 const XlaCompiler::CompilationResult* compilation_result; in TEST() local
75 &compilation_result, &executable); in TEST()
|
| D | get_compiler_ir.cc | 41 const XlaCompiler::CompilationResult* compilation_result = nullptr; in GetLocalExecutable() local
45 &compilation_result, &executable)); in GetLocalExecutable()
|
| D | xla_compilation_cache.cc | 457 compile_fn(&compiler, &entry->compilation_result); in CompileImpl()
461 BuildExecutable(options, entry->compilation_result, &entry->executable); in CompileImpl()
496 *out_compilation_result = &entry->compilation_result; in CompileImpl()
|
| D | xla_compilation_cache.h | 159 XlaCompiler::CompilationResult compilation_result TF_GUARDED_BY(mu);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 201 const XlaCompiler::CompilationResult& compilation_result, in AddVariableUpdatesToCores() argument
214 if (resource_update_pos < compilation_result.resource_updates.size()) { in AddVariableUpdatesToCores()
216 compilation_result.resource_updates[resource_update_pos]; in AddVariableUpdatesToCores()
219 int pos = compilation_result.outputs.size() + resource_update_pos; in AddVariableUpdatesToCores()
221 compilation_result.xla_output_shape, pos); in AddVariableUpdatesToCores()
271 const XlaCompiler::CompilationResult& compilation_result, in ComputeOutputShapesForEachCore() argument
275 TF_RET_CHECK(!compilation_result.outputs[i].is_constant) in ComputeOutputShapesForEachCore()
281 compilation_result.xla_output_shape, i); in ComputeOutputShapesForEachCore()
309 const tensorflow::XlaCompiler::CompilationResult& compilation_result, in CreateHloModules() argument
313 compilation_result.computation->proto().has_host_program_shape()); in CreateHloModules()
[all …]
|
| D | tpu_util.cc | 55 const XlaCompiler::CompilationResult& compilation_result) { in BuildAotXlaComputationInstance() argument
57 instance.computation = compilation_result.computation.get(); in BuildAotXlaComputationInstance()
58 for (const xla::Shape& shape : compilation_result.xla_input_shapes) { in BuildAotXlaComputationInstance()
61 instance.result_layout = &compilation_result.xla_output_shape; in BuildAotXlaComputationInstance()
|
| D | tpu_compile_op_support.h | 125 const XlaCompiler::CompilationResult& compilation_result,
133 const XlaCompiler::CompilationResult& compilation_result,
138 const XlaCompiler::CompilationResult& compilation_result,
|
| D | tpu_compile_op.cc | 47 const Tensor compilation_result = ctx->input(0); in Compute() local
50 if (!proto.ParseFromString(compilation_result.scalar<tstring>()())) { in Compute()
|
| D | tpu_util.h | 47 const XlaCompiler::CompilationResult& compilation_result);
|
| D | tpu_compile_op_common.h | 182 XlaCompiler::CompilationResult* compilation_result);
|
| D | tpu_compile_op_common.cc | 287 XlaCompiler::CompilationResult* compilation_result) { in CompileTFFunctionToHlo() argument
386 args, compilation_result); in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 79 const XlaCompiler::CompilationResult* compilation_result, in XlaExecutableClosure() argument
83 compilation_result_(compilation_result), in XlaExecutableClosure()
92 const XlaCompiler::CompilationResult* compilation_result() const { in compilation_result() function in tensorflow::__anon46563ba90111::XlaExecutableClosure
171 const XlaCompiler::CompilationResult** compilation_result, in CompileToLocalExecutable() argument
216 compilation_result, executable); in CompileToLocalExecutable()
225 const XlaCompiler::CompilationResult* compilation_result; in Compute() local
237 /*may_alias_resource_update=*/true, &client, &compilation_result, in Compute()
264 launch_context.PopulateInputs(ctx, compilation_result, resource_var_ptrs, in Compute()
293 ctx, compilation_result, execution_output->ConsumeResult(), in Compute()
492 closure.compilation_result()->xla_input_shapes.size(), ")"); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | BUILD | 65 "compilation_result.proto",
|
| /external/mesa3d/src/broadcom/compiler/ |
| D | vir_to_qpu.c | 420 c->compilation_result = V3D_COMPILATION_FAILED; in v3d_vir_to_qpu()
|
| D | vir.c | 527 c->compilation_result = V3D_COMPILATION_SUCCEEDED; in vir_compile_init()
1240 c->compilation_result != in v3d_compile()
|
| D | v3d_compiler.h | 725 enum v3d_compilation_result compilation_result; member
|
| D | nir_to_vir.c | 3439 c->compilation_result = in v3d_nir_to_vir()
|