Searched refs:IsDim0SliceAligned (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_util_test.cc | 323 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
329 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
343 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
359 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
367 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
383 bool output = IsDim0SliceAligned<float>(t.shape(), start, end); in TEST_F()
396 bool output = IsDim0SliceAligned<float>(shape, start, end); in TEST_F()
404 bool output = IsDim0SliceAligned<float>(shape, start, end); in TEST_F()
|
| D | slice_op.cc | 146 IsDim0SliceAligned<T>(input.shape(), (*begin)[0], (*size)[0])) { in SharedSliceCommonCases()
|
| D | strided_slice_op.cc | 120 if (slice_dim0 && IsDim0SliceAligned<T>(input.shape(), begin[0], end[0])) { in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | ops_util.h | 66 bool IsDim0SliceAligned(const TensorShape& s, int64 start, int64 end_or_size) { in IsDim0SliceAligned() function
|