Home
last modified time | relevance | path

Searched refs:slice_shape_ptrs (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2956 std::vector<const Shape*> slice_shape_ptrs; in AllToAllTuple() local
2957 absl::c_transform(slice_shapes, std::back_inserter(slice_shape_ptrs), in AllToAllTuple()
2960 Shape shape, ShapeInference::InferAllToAllTupleShape(slice_shape_ptrs)); in AllToAllTuple()