Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dops_util.h48 bool IsInnerDimsSizeAligned(const TensorShape& s) { in IsInnerDimsSizeAligned() function
79 return IsInnerDimsSizeAligned<T>(s); in IsDim0SliceAligned()
/external/tensorflow/tensorflow/core/kernels/
Dunpack_op.cc78 (output_size == 0 || IsInnerDimsSizeAligned<T>(input_shape))) { in Compute()
Dsplit_op.cc93 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) { in ComputeEasyCases()
Dsplit_v_op.cc149 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) { in ComputeEasyCases()
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dconcat_split_util.h142 if (IsInnerDimsSizeAligned<T>(input.shape())) { in SplitEasyCases()