Searched refs:bias_dim (Results 1 – 5 of 5) sorted by relevance
457 bias_dim = -1459 bias_dim = 2 if stacked_value else 1460 bias_shape = [x_shape[bias_dim]]
824 auto bias_dim = filter_type.getShape().back(); in matchAndRewrite() local826 RankedTensorType::get({bias_dim}, filter_type.getElementType()); in matchAndRewrite()913 auto bias_dim = filter_shape[2] * filter_shape[3]; in matchAndRewrite() local915 RankedTensorType::get({bias_dim}, filter_type.getElementType()); in matchAndRewrite()
362 auto bias_dim = static_cast<const ConcreteType *>(this)->getBiasDim( in matchAndRewrite() local364 auto bias_type = RankedTensorType::get({bias_dim}, elem_type); in matchAndRewrite()
397 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in BiasAddShape() local418 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()430 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()
1601 bias_dim = %filter.shape[2] * %filter.shape[3]1604 %zero_bias = tosa.CONST() {value={0} * bias_dim} () -> tensor<{bias_dim}, %input.dtype>