Searched refs:argument_shapes (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 266 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig() argument
271 if (program_shape.parameters_size() != argument_shapes.size()) { in CreateModuleConfig()
274 argument_shapes.size()); in CreateModuleConfig()
276 for (int i = 0; i < argument_shapes.size(); ++i) { in CreateModuleConfig()
279 if (!ShapeUtil::Compatible(*argument_shapes[i], in CreateModuleConfig()
285 ShapeUtil::HumanString(*argument_shapes[i])); in CreateModuleConfig()
289 *argument_shapes[i])); in CreateModuleConfig()
339 std::vector<const Shape*> argument_shapes; in CreateModuleConfig() local
341 argument_shapes.push_back(&arg->on_host_shape()); in CreateModuleConfig()
343 return CreateModuleConfig(program_shape, argument_shapes, &execution_options); in CreateModuleConfig()
[all …]
|
| D | service.h | 225 absl::Span<const Shape* const> argument_shapes,
|
| D | shape_inference.cc | 2838 string argument_shapes = in InferCallShape() local
2847 argument_shapes); in InferCallShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 96 def compile(self, computation, argument_shapes, result_shape, argument
142 def compile(self, c_computation, argument_shapes, result_shape, argument
144 return c_computation.Compile(argument_shapes, compile_options, self.client)
184 def compile(self, c_computation, argument_shapes, result_shape, argument
187 c_computation.GetSerializedProto(), argument_shapes, result_shape,
748 def Compile(self, argument_shapes=(), compile_options=None, layout_fn=None, argument
770 argument_shapes = [
771 shape.map_leaves(layout_fn) for shape in argument_shapes
775 argument_shapes = list(argument_shapes)
779 c = backend.compile(self.computation, argument_shapes, result_shape,
[all …]
|
| D | xrt.h | 88 const string& hlo_module_proto, const std::vector<Shape>& argument_shapes,
|
| D | xrt.cc | 145 const string& hlo_module_proto, const std::vector<Shape>& argument_shapes, in CompileForXrt() argument
155 for (auto& shape : argument_shapes) { in CompileForXrt()
|
| D | local_computation_builder.cc | 340 const std::vector<Shape>& argument_shapes, in Compile() argument
343 argument_shape_pointers.reserve(argument_shapes.size()); in Compile()
344 for (auto& argument_shape : argument_shapes) { in Compile()
|
| D | local_computation_builder.h | 161 const std::vector<Shape>& argument_shapes,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 63 absl::Span<const Shape> argument_shapes,
|
| D | client.cc | 218 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile() argument
235 for (const auto& arg_shape : argument_shapes) { in Compile()
|