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

  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 80 // T is the element type of the conv_input, filter and side_input tensors.
172 // The conv_input tensor is one of the following formats:
174 const Tensor& conv_input = context->input(kConvInput); variable
175 OP_REQUIRES_OK(context, CheckShape(conv_input, "conv_input"));
207 const int32 batch_size = GetTensorDim(conv_input, data_format_, 'N');
208 const int32 conv_input_rows = GetTensorDim(conv_input, data_format_, 'H');
209 const int32 conv_input_cols = GetTensorDim(conv_input, data_format_, 'W');
240 launcher_.launch(context, cudnn_use_autotune_, conv_input, conv_input_scale,
483 const Tensor* conv_input = &conv_input_param local
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 1174 auto& conv_input = op_context.op_info.inputs(0); local
    [all...]

Completed in 967 milliseconds