HomeSort by relevance Sort by last modified time
    Searched refs:major_dims (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/lib/
slicing.cc 32 auto major_dims = AsInt64Slice(shape.dimensions()) local
40 padded_start.begin() + major_dims.size());
44 std::copy(major_dims.begin(), major_dims.end(), padded_end.begin());
45 std::copy(end.begin(), end.end(), padded_end.begin() + major_dims.size());
118 auto major_dims = AsInt64Slice(shape.dimensions())
123 auto padded_sizes = ConcatVectors(major_dims, sizes);
matrix.cc 60 absl::Span<const int64> major_dims = local
65 auto mask = Broadcast(indicator, major_dims);
99 absl::Span<const int64> major_dims = local
105 return Broadcast(indicator, major_dims);
  /external/tensorflow/tensorflow/compiler/xla/service/
cholesky_expander.cc 60 auto major_dims = AsInt64Slice(a_shape.dimensions()) local
71 std::vector<int64> row_shape_dims(major_dims.begin(), major_dims.end());
72 std::vector<int64> col_shape_dims(major_dims.begin(), major_dims.end());

Completed in 635 milliseconds