Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc82 TensorShape input_tf_shape = input_mkl_shape.IsMklTensor() in ValidateMklInputs() local
85 const int input_dims = input_tf_shape.dims(); in ValidateMklInputs()
103 (*size)[i] = input_tf_shape.dim_size(i) - (*begin)[i]; in ValidateMklInputs()
111 if (input_tf_shape.dim_size(i) == 0) { in ValidateMklInputs()
118 OP_REQUIRES(context, 0 <= b && b <= input_tf_shape.dim_size(i), in ValidateMklInputs()
120 input_tf_shape.dim_size(i), in ValidateMklInputs()
122 OP_REQUIRES(context, 0 <= s && b + s <= input_tf_shape.dim_size(i), in ValidateMklInputs()
124 input_tf_shape.dim_size(i) - b, in ValidateMklInputs()
127 const bool take_all = (b == 0) && (s == input_tf_shape.dim_size(i)); in ValidateMklInputs()
Dmkl_conv_ops.h495 auto input_tf_shape = GetTfShape(context_, src_index); in GetOutputAndPadSizeInMklOrder() local
500 OP_REQUIRES(context_, input_tf_shape.dims() == 4, in GetOutputAndPadSizeInMklOrder()
502 input_tf_shape.DebugString())); in GetOutputAndPadSizeInMklOrder()
505 OP_REQUIRES(context_, input_tf_shape.dims() == 5, in GetOutputAndPadSizeInMklOrder()
507 input_tf_shape.DebugString())); in GetOutputAndPadSizeInMklOrder()
510 GetOutputAndPadSizeInMklOrder(input_tf_shape, filter_tf_shape, strides, in GetOutputAndPadSizeInMklOrder()
Dmkl_conv_grad_input_ops.cc524 TensorShape input_tf_shape; in MakeInputTfShape() local
529 TF_CHECK_OK(tensor::MakeShape(input_tensor, &input_tf_shape)); in MakeInputTfShape()
530 return input_tf_shape; in MakeInputTfShape()