Home
last modified time | relevance | path

Searched refs:IsInnerDimsSizeAligned (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dops_util.h49 bool IsInnerDimsSizeAligned(const TensorShape& s) { in IsInnerDimsSizeAligned() function
80 return IsInnerDimsSizeAligned<T>(s); in IsDim0SliceAligned()
/external/tensorflow/tensorflow/core/kernels/
Dunpack_op.cc83 (output_size == 0 || IsInnerDimsSizeAligned<T>(input_shape))) { in Compute()
Dsplit_op.cc96 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) { in ComputeEasyCases()
Dsplit_v_op.cc149 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) { in ComputeEasyCases()
Dbatch_kernels.cc127 if (IsInnerDimsSizeAligned<T>(input.shape())) { in SplitEasyCases()