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

  /external/tensorflow/tensorflow/core/kernels/
unpack_op.cc 93 int64 before_dim = 1; variable
95 before_dim *= input_shape.dim_size(i);
107 input.shaped<T, 3>({1, before_dim, axis_dim * after_dim});
115 auto output_shaped = output->shaped<T, 3>({1, before_dim, after_dim});
117 Eigen::DSizes<Eigen::DenseIndex, 3> sizes{1, before_dim, after_dim};
pack_op.cc 90 int64 before_dim = 1; variable
92 before_dim *= output_shape.dim_size(i);
105 output->shaped<T, 2>({before_dim, after_dim * axis_dim});
113 values[i].shaped<T, 2>({before_dim, after_dim})));

Completed in 83 milliseconds