Searched refs:IsDim0SliceAligned (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_util_test.cc | 325 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
331 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
345 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
361 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
369 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
385 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
398 bool output = IsDim0SliceAligned<float>(shape, start, end); in TEST_F()
406 bool output = IsDim0SliceAligned<float>(shape, start, end); in TEST_F()
|
| D | slice_op.cc | 143 IsDim0SliceAligned<T>(input.shape(), (*begin)[0], (*size)[0])) { in SharedSliceCommonCases()
|
| D | strided_slice_op.cc | 122 if (slice_dim0 && IsDim0SliceAligned<T>(input.shape(), begin[0], end[0])) { in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | ops_util.h | 65 bool IsDim0SliceAligned(const TensorShape& s, int64 start, int64 end_or_size) { in IsDim0SliceAligned() function
|