HomeSort by relevance Sort by last modified time
    Searched refs:IsInnerDimsSizeAligned (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/framework/
ops_util.h 49 bool IsInnerDimsSizeAligned(const TensorShape& s) {
80 return IsInnerDimsSizeAligned<T>(s);
  /external/tensorflow/tensorflow/core/kernels/
unpack_op.cc 83 (output_size == 0 || IsInnerDimsSizeAligned<T>(input_shape))) {
split_op.cc 96 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) {
split_v_op.cc 149 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) {
batch_kernels.cc 127 if (IsInnerDimsSizeAligned<T>(input.shape())) {
    [all...]

Completed in 455 milliseconds