Searched refs:ConcatVectors (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 38 std::vector<int64> ConcatVectors(absl::Span<const int64> xs, in ConcatVectors() function
214 vs, Reshape(v, ConcatVectors(batch_dims, {m, 1})), {j}); in QRBlock()
217 taus, Reshape(tau, ConcatVectors(batch_dims, {1})), {j}); in QRBlock()
222 builder, ShapeUtil::MakeShape(type, ConcatVectors(batch_dims, {m, n}))); in QRBlock()
224 ShapeUtil::MakeShape(type, ConcatVectors(batch_dims, {n}))); in QRBlock()
280 /*broadcast_dimensions=*/ConcatVectors(batch_dim_indices, {n_index})); in ComputeWYRepresentation()
290 ShapeUtil::MakeShape(type, ConcatVectors(batch_dims, {m, n}))); in ComputeWYRepresentation()
297 /*broadcast_dimensions=*/ConcatVectors(batch_dim_indices, {n_index})); in ComputeWYRepresentation()
|
| D | slicing.cc | 86 std::vector<int64> ConcatVectors(absl::Span<const int64> xs, in ConcatVectors() function
123 auto padded_sizes = ConcatVectors(major_dims, sizes); in DynamicSliceInMinorDims()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | util.h | 42 std::vector<int64> ConcatVectors(absl::Span<const int64> xs,
|
| D | util.cc | 123 std::vector<int64> ConcatVectors(absl::Span<const int64> xs, in ConcatVectors() function
|
| /external/google-fruit/include/fruit/impl/meta/ |
| D | set.h | 157 using SetUncheckedUnion = ConcatVectors;
|
| D | vector.h | 152 struct ConcatVectors {
|
| /external/google-fruit/include/fruit/impl/ |
| D | component_functors.defn.h | 692 using new_Deps = ConcatVectors(typename OtherComp::Deps, typename Comp::Deps);
|