Searched refs:ConcatVectors (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 40 std::vector<int64> ConcatVectors(absl::Span<const int64> xs, in ConcatVectors() function
211 builder, ShapeUtil::MakeShape(S32, ConcatVectors(batch_dims, {m, n})), in QrBlock()
241 /*broadcast_dimensions=*/ConcatVectors(batch_dim_indices, {minor_dim})); in QrBlock()
245 new_x = BroadcastInDim(new_x, ConcatVectors(batch_dims, {m, n}), in QrBlock()
254 Iota(builder, ShapeUtil::MakeShape(S32, ConcatVectors(batch_dims, {n})), in QrBlock()
267 ShapeUtil::MakeShape(type, ConcatVectors(batch_dims, {std::min(m, n)}))); in QrBlock()
320 auto tau_scale = BroadcastInDim(-taus, ConcatVectors(batch_dims, {1, n}), in CompactWYRepresentation()
321 ConcatVectors(batch_dim_indices, {n_index})); in CompactWYRepresentation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | util.h | 42 std::vector<int64> ConcatVectors(absl::Span<const int64> xs,
|
| D | util.cc | 126 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/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 107 std::vector<int64> ConcatVectors(absl::Span<const int64> xs, in ConcatVectors() function
144 auto padded_sizes = ConcatVectors(major_dims, sizes); in DynamicSliceInMinorDims()
|
| /external/google-fruit/include/fruit/impl/ |
| D | component_functors.defn.h | 692 using new_Deps = ConcatVectors(typename OtherComp::Deps, typename Comp::Deps);
|