Home
last modified time | relevance | path

Searched defs:parameter_shapes (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_proto_util.cc65 std::vector<const ShapeProto*> parameter_shapes; in EntryComputationParameterShapes() local
H A Dhlo_computation.cc764 std::vector<Shape> parameter_shapes(instruction->operand_count()); in CreateAsyncInstructions() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_lowering_context.h29 const std::vector<Shape>& parameter_shapes() const override { in parameter_shapes() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dparams_test.cc356 std::vector<Shape> parameter_shapes; in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc1780 std::vector<Shape> parameter_shapes; in PjRtStreamExecutorExecutable() local