Searched defs:argument_shapes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 119 LocalExecutable::RunHelper(const absl::Span<const Shape* const> argument_shapes, in RunHelper()
175 std::vector<const Shape*> argument_shapes; in Run() local
188 std::vector<const Shape*> argument_shapes; in Run() local
243 std::vector<const Shape*> argument_shapes; in RunAsync() local
324 std::vector<const Shape*> argument_shapes; in RunAsync() local
|
| D | compile_only_client.cc | 28 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
|
| D | local_client.h | 103 absl::Span<Shape const* const> argument_shapes, in AsyncCallAndBlockHostUntilDone()
|
| D | client.cc | 218 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_util.cc | 46 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
|
| D | service.cc | 257 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
277 std::vector<const Shape*> argument_shapes; in CreateModuleConfig() local
804 std::vector<Shape> argument_shapes; in Compile() local
|
| D | shape_inference.cc | 3183 string argument_shapes = in InferCallShape() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 57 const ProgramShape& program_shape, absl::Span<const Shape> argument_shapes, in CreateModuleConfig()
136 absl::Span<const Shape> argument_shapes, in CreateModuleConfig()
|