Searched defs:argument_shapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 266 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
339 std::vector<const Shape*> argument_shapes; in CreateModuleConfig() local
828 std::vector<Shape> argument_shapes; in Compile() local
|
| D | shape_inference.cc | 2838 string argument_shapes = in InferCallShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xrt.cc | 145 const string& hlo_module_proto, const std::vector<Shape>& argument_shapes, in CompileForXrt()
|
| 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
184 def compile(self, c_computation, argument_shapes, result_shape, argument
748 def Compile(self, argument_shapes=(), compile_options=None, layout_fn=None, argument
|
| D | local_computation_builder.cc | 340 const std::vector<Shape>& argument_shapes, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 218 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile()
|