Searched refs:StridedSliceAssignBCast (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op_test.cc | 28 using Vec = typename StridedSliceAssignBCast::Vec;
98 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
118 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
137 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
160 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
198 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
215 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
240 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
264 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
292 StridedSliceAssignBCast bcast(test_pair.from, test_pair.to); in TEST()
[all …]
|
| H A D | strided_slice_op.cc | 450 StridedSliceAssignBCast::StridedSliceAssignBCast( in StridedSliceAssignBCast() function in tensorflow::StridedSliceAssignBCast
451 const StridedSliceAssignBCast::Vec& input_shape, in StridedSliceAssignBCast()
452 const StridedSliceAssignBCast::Vec& output_shape) in StridedSliceAssignBCast()
496 bool StridedSliceAssignBCast::RemapDimensions( in RemapDimensions()
497 int64_t num_dims, const StridedSliceAssignBCast::Vec& dimension_map) { in RemapDimensions()
|
| H A D | strided_slice_op.h | 96 class StridedSliceAssignBCast {
100 StridedSliceAssignBCast(const Vec& input_shape, const Vec& output_shape);
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op_impl.h | 62 const StridedSliceAssignBCast& bcast, Tensor* result);
145 const StridedSliceAssignBCast& bcast, Tensor* result) { in operator()
172 const StridedSliceAssignBCast& bcast, Tensor* result) { in operator()
222 const StridedSliceAssignBCast& bcast); \
|
| H A D | strided_slice_op.h | 96 const StridedSliceAssignBCast& bcast) {
|
| H A D | strided_slice_op.cc | 358 StridedSliceAssignBCast bcast(input_shape.dim_sizes(), in Compute()
|