Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc470 std::vector<Shape> element_shapes; in MakeTuple() local
483 std::vector<Shape> element_shapes; in MakeTupleFromSlices() local
496 std::vector<Shape> element_shapes; in MakeTupleOwned() local
Dliteral.cc1552 std::vector<Shape> element_shapes; in MoveIntoTuple() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc415 std::vector<Shape> element_shapes(kElementCount, element_shape); in XLA_TEST_F() local
455 std::vector<Shape> element_shapes(kFanout, element_shape); in XLA_TEST_F() local
/external/tensorflow/tensorflow/core/kernels/
Dtensor_array_ops.cc1278 std::vector<TensorShape> element_shapes(num_tensors, tensor_value->shape()); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc1607 std::vector<Shape> element_shapes; in CreateTuple() local