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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc 41 auto input_feature_dim = dnums.input_feature_dimension(); local
68 new_input_dim_order[num_dims - 1] = input_feature_dim;
70 input->shape().dimensions(input_feature_dim);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_op_helpers.cc 130 int64 input_feature_dim = filter_shape.dimensions_size() - 2; local
133 int64 input_feature = filter_shape.dimensions(input_feature_dim);
137 implicit_broadcast_filter_shape.set_dimensions(input_feature_dim, 1);
  /external/tensorflow/tensorflow/compiler/xla/service/
convolution_group_converter.cc 105 int64 input_feature_dim) {
110 input_feature_dim, shape.dimensions(input_feature_dim) * group_count);
356 // We want to repeat 'filter' in the 'input_feature_dim' dimension
hlo_cost_analysis.cc 445 const int64 input_feature_dim = dnums.input_feature_dimension(); local
448 ShapeUtil::GetDimension(lhs_shape, input_feature_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_rewriter.cc 119 auto input_feature_dim = conv_dnums.input_feature_dimension(); local
230 backward_conv_dnums.set_input_batch_dimension(input_feature_dim);

Completed in 330 milliseconds