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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h 282 auto input_tf_shape = GetTfShape(context_, src_index); local
285 OP_REQUIRES(context_, input_tf_shape.dims() == 4,
287 input_tf_shape.DebugString()));
289 GetOutputAndPadSizeInMklOrder(input_tf_shape, filter_tf_shape, strides,
380 TensorShape input_tf_shape = MakeInputTfShape(context, input_tensor); variable
386 if (input_tf_shape.num_elements() == 0 ||
392 input_tf_shape, filter_tf_shape, outbprop_tf_shape);
415 input_tf_shape, filter_tf_shape, &fwd_input_dims, &fwd_filter_dims,
mkl_conv_grad_input_ops.cc 386 TensorShape input_tf_shape; local
389 TensorShapeUtils::MakeShape(input_tensor.vec<int32>(), &input_tf_shape)
392 return input_tf_shape;

Completed in 108 milliseconds