Home
last modified time | relevance | path

Searched refs:target_shape_in (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreshape_util.cc35 const Tensor &input_shape_in, const Tensor &target_shape_in, in Reshape() argument
45 OP_REQUIRES(context, TensorShapeUtils::IsVector(target_shape_in.shape()), in Reshape()
48 target_shape_in.shape().DebugString())); in Reshape()
51 const int64 output_rank = target_shape_in.NumElements(); in Reshape()
61 auto target_shape = target_shape_in.vec<int64>(); in Reshape()
Dreshape_util.h26 const Tensor &input_shape_in, const Tensor &target_shape_in,