Home
last modified time | relevance | path

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

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