| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bias_ops.cc | 54 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
|
| D | conv_op_helpers.cc | 107 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
210 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
301 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
|
| D | depthtospace_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
|
| D | extract_image_patches_op.cc | 62 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile() local
|
| D | spacetodepth_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | approx_topk_test.py | 119 def test_mips(self, k, db_size, qy_size, feature_dim): argument
146 def test_l2ann(self, k, db_size, qy_size, feature_dim): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | approx_topk_test.py | 121 def test_mips(self, dtype, k, db_size, qy_size, feature_dim): argument
147 def test_l2ann(self, dtype, k, db_size, qy_size, feature_dim): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 125 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_simplify_padding.cc | 92 int64_t feature_dim = dnums.kernel_output_feature_dimension(); in NumTrailingZeroOutputFeatures() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 385 DimensionHandle feature_dim = c->Dim(c->input(3), 1); in __anon0a5055ea0a02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 408 Value broadcast_from, int64_t feature_dim, in Broadcast1DToFeatureDim()
991 int64_t feature_dim = GetTensorFeatureDimIndex(num_dims, format); in GetConvDimensionNumbersAttr() local
1028 auto feature_dim = GetFeatureDimension(data_format, value_type); in matchAndRewrite() local
2236 auto feature_dim = feature_dim_attr.getValue().getSExtValue(); in matchAndRewrite() local
2345 auto feature_dim = getFeatureDimensionAttr(rewriter, data_format, op.x()); in matchAndRewrite() local
4998 const int feature_dim = in matchAndRewrite() local
5161 const int feature_dim = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 500 bool NeedsReformatTypeAndPermutation(int batch_dim, int feature_dim, in NeedsReformatTypeAndPermutation()
513 GetReformatTypeAndPermutation(int batch_dim, int feature_dim, in GetReformatTypeAndPermutation()
539 Value FormatToNHWC(Value value, int batch_dim, int feature_dim, in FormatToNHWC()
|
| D | lower_tf.cc | 140 int64_t feature_dim = GetTensorFeatureDimIndex(rank, format); in GetBiasAddGradReductionIndices() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 528 int64_t feature_dim = value_ty.getDimSize(feature_dim_idx); in verify() local
|