Searched defs:argument_shapes (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.cc | 120 LocalExecutable::RunHelper(const absl::Span<const Shape* const> argument_shapes, in RunHelper()
178 std::vector<const Shape*> argument_shapes; in Run() local
191 std::vector<const Shape*> argument_shapes; in Run() local
246 std::vector<const Shape*> argument_shapes; in RunAsync() local
327 std::vector<const Shape*> argument_shapes; in RunAsync() local
|
| H A D | compile_only_client.cc | 29 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
|
| H A D | local_client.h | 104 absl::Span<Shape const* const> argument_shapes, in AsyncCallAndBlockHostUntilDone()
|
| H A D | client.cc | 217 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_support.cc | 59 const ProgramShape& program_shape, absl::Span<const Shape> argument_shapes, in CreateModuleConfig()
138 absl::Span<const Shape> argument_shapes, in CreateModuleConfig()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.cc | 262 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
282 std::vector<const Shape*> argument_shapes; in CreateModuleConfig() local
867 std::vector<Shape> argument_shapes; in Compile() local
|
| H A D | hlo_module_util.cc | 46 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig()
|
| H A D | shape_inference.cc | 3302 std::string argument_shapes = in InferCallShape() local
|