Home
last modified time | relevance | path

Searched refs:input_shape_in (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreshape_util.cc35 const Tensor &input_shape_in, const Tensor &target_shape_in, in Reshape() argument
41 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_shape_in.shape()), in Reshape()
44 input_shape_in.shape().DebugString())); in Reshape()
50 const int64 input_rank = input_shape_in.NumElements(); in Reshape()
52 const TensorShape input_shape(input_shape_in.vec<int64>()); in Reshape()
105 context->set_output(output_shape_idx, input_shape_in); in Reshape()
Dsparse_reorder_op.cc51 const Tensor& input_shape_in = context->input(2); in Compute() local
52 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_shape_in.shape()), in Compute()
55 input_shape_in.shape().DebugString())); in Compute()
57 const TensorShape input_shape(input_shape_in.vec<int64>()); in Compute()
Dreshape_util.h26 const Tensor &input_shape_in, const Tensor &target_shape_in,