| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tf_xla_mlir_translate.cc | 83 const XlaCompilationResult& compilation_result, llvm::raw_ostream& output) { in PrintHloModuleText() argument
85 compilation_result.computation->GetProgramShape().ValueOrDie()); in PrintHloModuleText()
87 compilation_result.computation->proto(), module_config); in PrintHloModuleText()
98 if (!compilation_result.input_mapping.empty()) in PrintHloModuleText()
100 << absl::StrJoin(compilation_result.input_mapping, ", ") << "}\n"; in PrintHloModuleText()
102 for (const auto& xla_input_shape : compilation_result.xla_input_shapes) in PrintHloModuleText()
106 << compilation_result.xla_output_shape.ToString() << '\n'; in PrintHloModuleText()
108 for (const auto& xla_output_description : compilation_result.outputs) { in PrintHloModuleText()
120 for (const auto& resource_update : compilation_result.resource_updates) { in PrintHloModuleText()
256 llvm::StringRef device_type, XlaCompilationResult* compilation_result, in CompileMlirToXlaHloViaBuilder() argument
[all …]
|
| D | compile_mlir_util.h | 114 XlaCompilationResult* compilation_result);
126 XlaCompilationResult* compilation_result,
136 XlaCompilationResult* compilation_result,
150 XlaCompilationResult* compilation_result,
162 XlaCompilationResult* compilation_result,
|
| D | compile_mlir_util.cc | 588 XlaCompilationResult* compilation_result) { in PopulateCollectiveInfo() argument
610 compilation_result->collective_info = {group_key, group_size, 0}; in PopulateCollectiveInfo()
618 XlaCompilationResult* compilation_result) { in PopulateResultIOInfo() argument
621 GetInputMappingForMlir(arg_shapes.size(), &compilation_result->input_mapping); in PopulateResultIOInfo()
626 &compilation_result->xla_input_shapes)); in PopulateResultIOInfo()
631 &compilation_result->xla_output_shape, &compilation_result->outputs, in PopulateResultIOInfo()
632 &compilation_result->resource_updates); in PopulateResultIOInfo()
640 XlaCompilationResult* compilation_result, in CompileMlirToXlaHlo() argument
652 compilation_result->computation = std::make_shared<xla::XlaComputation>(); in CompileMlirToXlaHlo()
654 module_op, device_type, compilation_result->computation.get(), in CompileMlirToXlaHlo()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 249 const XlaCompiler::CompilationResult* compilation_result, in PopulateInputs() argument
254 arguments.reserve(compilation_result->xla_input_shapes.size()); in PopulateInputs()
256 for (int i = 0, end = compilation_result->xla_input_shapes.size(); i < end; in PopulateInputs()
258 int arg_num = compilation_result->input_mapping[i]; in PopulateInputs()
260 const xla::Shape& device_shape = compilation_result->xla_input_shapes[i]; in PopulateInputs()
267 absl::c_any_of(compilation_result->resource_updates, in PopulateInputs()
380 const XlaCompiler::CompilationResult* compilation_result, int output_num) { in SetOutputForConstant() argument
381 CHECK(compilation_result->outputs[output_num].is_constant); in SetOutputForConstant()
383 compilation_result->outputs[output_num].constant_value; in SetOutputForConstant()
435 const XlaCompiler::CompilationResult& compilation_result, in GatherVariableInfo() argument
[all …]
|
| D | xla_compilation_cache_disable_test.cc | 46 const XlaCompiler::CompilationResult* compilation_result; in TEST() local
57 &compilation_result, &executable); in TEST()
66 &compilation_result, &executable); in TEST()
75 &compilation_result, &executable); in TEST()
|
| D | xla_compilation_cache.cc | 397 XlaCompiler::CompilationResult* compilation_result) { in XlaSingleOpToHlo() argument
406 *compilation_result = {}; in XlaSingleOpToHlo()
408 std::move(graph), args, compilation_result); in XlaSingleOpToHlo()
437 options.shape_determination_fns, compilation_result); in XlaSingleOpToHlo()
507 compile_options, &entry->compilation_result); in CompileStrict()
512 &entry->compilation_result); in CompileStrict()
517 TF_RET_CHECK(entry->compilation_result.computation != nullptr); in CompileStrict()
522 entry->compilation_result.computation->proto(); in CompileStrict()
541 options, entry->compilation_result, serialized_entry->executable()); in CompileStrict()
548 BuildExecutable(options, entry->compilation_result, &entry->executable); in CompileStrict()
[all …]
|
| D | xla_launch_util.h | 94 const XlaCompiler::CompilationResult& compilation_result,
166 const XlaCompiler::CompilationResult* compilation_result,
183 const XlaCompiler::CompilationResult* compilation_result,
|
| D | xla_compilation_cache.h | 217 XlaCompiler::CompilationResult compilation_result TF_GUARDED_BY(mu);
331 XlaCompiler::CompilationResult* compilation_result);
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 81 const XlaCompiler::CompilationResult* compilation_result, in XlaExecutableClosure() argument
85 compilation_result_(compilation_result), in XlaExecutableClosure()
94 const XlaCompiler::CompilationResult* compilation_result() const { in compilation_result() function in tensorflow::__anonad8d49c70111::XlaExecutableClosure
244 const XlaCompiler::CompilationResult** compilation_result, in CompileToLocalExecutable() argument
280 compilation_result, executable); in CompileToLocalExecutable()
311 const XlaCompiler::CompilationResult* compilation_result; in ComputeAsync() local
325 /*may_alias_resource_update=*/true, &client, &compilation_result, in ComputeAsync()
331 compilation_result, done, inputs, in ComputeAsync()
349 launch_context.PopulateInputs(ctx, compilation_result, snapshot_ptrs, in ComputeAsync()
357 if (compilation_result->collective_info.has_value()) { in ComputeAsync()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 203 const XlaCompiler::CompilationResult& compilation_result, in AddVariableUpdatesToCores() argument
216 if (resource_update_pos < compilation_result.resource_updates.size()) { in AddVariableUpdatesToCores()
218 compilation_result.resource_updates[resource_update_pos]; in AddVariableUpdatesToCores()
221 int pos = compilation_result.outputs.size() + resource_update_pos; in AddVariableUpdatesToCores()
223 compilation_result.xla_output_shape, pos); in AddVariableUpdatesToCores()
276 const XlaCompiler::CompilationResult& compilation_result, in ComputeOutputShapesForEachCore() argument
280 TF_RET_CHECK(!compilation_result.outputs[i].is_constant) in ComputeOutputShapesForEachCore()
286 compilation_result.xla_output_shape, i); in ComputeOutputShapesForEachCore()
314 const tensorflow::XlaCompiler::CompilationResult& compilation_result, in CreateHloModules() argument
318 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);
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.h | 48 bool use_tuple_args, XlaCompiler::CompilationResult* compilation_result);
|
| D | tpu_compile.cc | 380 bool use_tuple_args, XlaCompiler::CompilationResult* compilation_result) { in CompileTFFunctionToHlo() argument
448 args, compilation_result); in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | BUILD | 55 "compilation_result.proto",
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 699 ArrayRef<TF::TPUCompilationResultOp> compilation_result, OpBuilder* builder, in Rewrite() argument
794 for (auto res : compilation_result) { in Rewrite()
|
| /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()
|