Searched refs:computation_proto (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 77 const xrt::XLAComputation& computation_proto,
85 const xrt::XLAComputation& computation_proto, in Compile() argument
88 computation_proto, xla_mesh_state, tpu_program_group); in Compile()
170 xrt::XLAComputation computation_proto; in Compute() local
173 computation_proto.ParseFromString(computation_input.scalar<tstring>()()), in Compute()
177 const xrt::XLAComputationConfig& config = computation_proto.config(); in Compute()
184 computation_proto, mesh_state, num_replicas, num_cores_per_replica); in Compute()
205 return Compile(mesh_state->data(), computation_proto, in Compute()
|
| D | xrt_compile_ops.cc | 106 const xrt::XLAComputation& computation_proto,
124 const xrt::XLAComputation& computation_proto, in Compile() argument
126 const xrt::XLAComputationConfig& config = computation_proto.config(); in Compile()
151 client->LoadSnapshot(computation_proto.hlo_snapshot())); in Compile()
199 xrt::XLAComputation computation_proto; in Compute() local
202 &computation_proto), in Compute()
207 OP_REQUIRES_OK(ctx, CompilationCacheKey(computation_proto, &key)); in Compute()
220 return Compile(ctx, computation_proto, program); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module.cc | 312 HloComputationProto computation_proto = computation->ToProto(); in ToProto() local
313 proto.add_computations()->Swap(&computation_proto); in ToProto()
426 for (const HloComputationProto& computation_proto : proto.computations()) { in CreateFromProto() local
429 HloComputation::CreateFromProto(computation_proto, computation_map, in CreateFromProto()
432 int64_t computation_id = computation_proto.id(); in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.cc | 604 const HloComputationProto* computation_proto = in AnalyzeConstantValueFallback() local
607 [root, computation_proto, context, in AnalyzeConstantValueFallback()
611 HloComputation::CreateFromProto(*computation_proto, {})); in AnalyzeConstantValueFallback()
979 const HloComputationProto* computation_proto = in AnalyzeConstant() local
982 [root, context, computation_proto, in AnalyzeConstant()
986 HloComputation::CreateFromProto(*computation_proto, {})); in AnalyzeConstant()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 201 for (const auto& computation_proto : computation->proto().computations()) { in CustomCallInternal() local
203 builder_.getContext(), computation_proto.name())); in CustomCallInternal()
|