Searched refs:input_batch_dim (Results 1 – 6 of 6) sorted by relevance
41 auto input_batch_dim = dnums.input_batch_dimension(); in Run() local62 new_input_dim_order[0] = input_batch_dim; in Run()63 new_input_dims[0] = input->shape().dimensions(input_batch_dim); in Run()
108 TfLiteIntArray* input_batch_dim = in ApplyDelegate() local113 input_batch_dim->data[i] = input_batch_dimensions[i]; in ApplyDelegate()119 options.input_batch_dimensions = input_batch_dim; in ApplyDelegate()
181 auto input_batch_dim = conv_dnums.input_batch_dimension(); in MatchBackwardFilter() local309 backward_conv_dnums.set_input_feature_dimension(input_batch_dim); in MatchBackwardFilter()
579 const int64 input_batch_dim = dnums.input_batch_dimension(); in HandleConvolution() local586 const int64 batch = ShapeUtil::GetDimension(lhs_shape, input_batch_dim); in HandleConvolution()
1130 const int64 input_batch_dim = dnums.input_batch_dimension();1143 ShapeUtil::GetDimension(lhs_shape, input_batch_dim);1227 lhs_dim_multipliers[input_batch_dim];1237 lhs_dim_multipliers[input_batch_dim];
2590 const int input_batch_dim = input_tensor.batch_size(); in ConvertReshape() local2600 VLOG(1) << "input_batch_dim=" << input_batch_dim in ConvertReshape()2608 if (input_batch_dim != -1 && output_batch_dim != -1) { in ConvertReshape()2609 reshape_may_change_batch_dim = (input_batch_dim != output_batch_dim); in ConvertReshape()2618 params->node_def.name(), ". input_batch_dim=", input_batch_dim, in ConvertReshape()