Home
last modified time | relevance | path

Searched refs:bias_dim (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py457 bias_dim = -1
459 bias_dim = 2 if stacked_value else 1
460 bias_shape = [x_shape[bias_dim]]
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc824 auto bias_dim = filter_type.getShape().back(); in matchAndRewrite() local
826 RankedTensorType::get({bias_dim}, filter_type.getElementType()); in matchAndRewrite()
913 auto bias_dim = filter_shape[2] * filter_shape[3]; in matchAndRewrite() local
915 RankedTensorType::get({bias_dim}, filter_type.getElementType()); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc362 auto bias_dim = static_cast<const ConcreteType *>(this)->getBiasDim( in matchAndRewrite() local
364 auto bias_type = RankedTensorType::get({bias_dim}, elem_type); in matchAndRewrite()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc397 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in BiasAddShape() local
418 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()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md1601 bias_dim = %filter.shape[2] * %filter.shape[3]
1604 %zero_bias = tosa.CONST() {value={0} * bias_dim} () -> tensor<{bias_dim}, %input.dtype>