| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 105 HloInstruction* operand, int64 feature_index, in DynamicElementCountPerFeature() argument
112 if (i == feature_index) { in DynamicElementCountPerFeature()
175 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormTraining() local
193 if (i != feature_index) { in HandleBatchNormTraining()
199 add(DynamicElementCountPerFeature(operand, feature_index, add)); in HandleBatchNormTraining()
202 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index})); in HandleBatchNormTraining()
205 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index})); in HandleBatchNormTraining()
227 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index})); in HandleBatchNormTraining()
241 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index})); in HandleBatchNormTraining()
301 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormInference() local
[all …]
|
| D | shape_inference.h | 91 int64 feature_index);
98 const Shape& variance_shape, int64 feature_index);
106 int64 feature_index);
|
| D | shape_inference.cc | 1212 const Shape& offset_shape, int64 feature_index) { in InferBatchNormTrainingShape() argument
1227 if (feature_index >= operand_shape.rank()) { in InferBatchNormTrainingShape()
1232 feature_index, operand_shape.rank()); in InferBatchNormTrainingShape()
1235 if (feature_index < 0) { in InferBatchNormTrainingShape()
1239 feature_index); in InferBatchNormTrainingShape()
1290 const int64 feature_count = operand_shape.dimensions(feature_index); in InferBatchNormTrainingShape()
1318 const Shape& variance_shape, int64 feature_index) { in InferBatchNormInferenceShape() argument
1333 if (feature_index >= operand_shape.rank()) { in InferBatchNormInferenceShape()
1338 feature_index, operand_shape.rank()); in InferBatchNormInferenceShape()
1341 if (feature_index < 0) { in InferBatchNormInferenceShape()
[all …]
|
| D | hlo_instructions.cc | 78 HloInstruction* scale, float epsilon, int64 feature_index) in HloBatchNormInstruction() argument
81 feature_index_(feature_index) { in HloBatchNormInstruction()
91 return feature_index() == casted_other.feature_index() && in IdenticalSlowPath()
105 StrCat("feature_index=", feature_index())}; in ExtraAttributesToStringImpl()
110 HloInstruction* offset, float epsilon, int64 feature_index) in HloBatchNormTrainingInstruction() argument
112 scale, epsilon, feature_index) { in HloBatchNormTrainingInstruction()
123 feature_index()); in CloneWithNewOperandsImpl()
129 float epsilon, int64 feature_index) in HloBatchNormInferenceInstruction() argument
131 scale, epsilon, feature_index) { in HloBatchNormInferenceInstruction()
144 new_operands[4], epsilon(), feature_index()); in CloneWithNewOperandsImpl()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 471 int64 feature_index; member
479 os << "feature_index=" << p.feature_index << ", "; in operator <<()
506 auto add_testcase = [&](std::vector<int64> bounds, int64 feature_index, in BuildBatchNormTestParams()
508 BatchNormTestParam p{bounds, feature_index, random_value_mean, in BuildBatchNormTestParams()
555 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
557 Product(bounds) / bounds[feature_index]; in XLA_TEST_P()
558 const int64 feature_bound = bounds[feature_index]; in XLA_TEST_P()
566 if (i != feature_index) { in XLA_TEST_P()
601 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
602 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
[all …]
|
| /external/harfbuzz_ng/src/ |
| D | hb-ot-layout.cc | 645 unsigned int *feature_index /* OUT */) in hb_ot_layout_table_find_feature() argument
654 if (feature_index) *feature_index = i; in hb_ot_layout_table_find_feature()
659 if (feature_index) *feature_index = HB_OT_LAYOUT_NO_FEATURE_INDEX; in hb_ot_layout_table_find_feature()
788 unsigned int *feature_index) in hb_ot_layout_language_get_required_feature_index() argument
794 feature_index, in hb_ot_layout_language_get_required_feature_index()
820 unsigned int *feature_index, in hb_ot_layout_language_get_required_feature() argument
827 if (feature_index) *feature_index = index; in hb_ot_layout_language_get_required_feature()
927 unsigned int *feature_index /* OUT */) in hb_ot_layout_language_find_feature() argument
938 if (feature_index) *feature_index = f_index; in hb_ot_layout_language_find_feature()
943 if (feature_index) *feature_index = HB_OT_LAYOUT_NO_FEATURE_INDEX; in hb_ot_layout_language_find_feature()
[all …]
|
| D | hb-ot-layout.h | 202 unsigned int *feature_index);
209 unsigned int *feature_index,
236 unsigned int *feature_index);
241 unsigned int feature_index,
313 unsigned int feature_index,
389 unsigned int feature_index,
400 unsigned int feature_index,
|
| D | hb-ot-map.cc | 100 unsigned int feature_index, in add_lookups() argument
118 feature_index, in add_lookups()
228 unsigned int feature_index[2]; in compile() local
239 &feature_index[table_index]); in compile()
248 &feature_index[table_index]); in compile()
258 map->index[0] = feature_index[0]; in compile()
259 map->index[1] = feature_index[1]; in compile()
|
| D | hb-ot-shape-complex-arabic-fallback.hh | 50 unsigned int feature_index) in arabic_fallback_synthesize_lookup_single() argument
59 hb_codepoint_t s = shaping_table[u - SHAPING_TABLE_FIRST][feature_index]; in arabic_fallback_synthesize_lookup_single()
179 unsigned int feature_index) in arabic_fallback_synthesize_lookup() argument
181 if (feature_index < 4) in arabic_fallback_synthesize_lookup()
182 return arabic_fallback_synthesize_lookup_single (plan, font, feature_index); in arabic_fallback_synthesize_lookup()
|
| /external/harfbuzz_ng/test/api/ |
| D | test-ot-name.c | 36 unsigned int feature_index; in test_ot_layout_feature_get_name_ids_and_characters() local
50 &feature_index)) in test_ot_layout_feature_get_name_ids_and_characters()
53 if (!hb_ot_layout_feature_get_name_ids (face, HB_OT_TAG_GSUB, feature_index, in test_ot_layout_feature_get_name_ids_and_characters()
64 all_chars = hb_ot_layout_feature_get_characters (face, HB_OT_TAG_GSUB, feature_index, in test_ot_layout_feature_get_name_ids_and_characters()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 85 int feature_index = in CompileImpl() local
95 input, ctx->Input(1), ctx->Input(2), epsilon_, feature_index); in CompileImpl()
176 epsilon_, feature_index); in CompileImpl()
268 const int feature_index = in Compile() local
293 epsilon_, feature_index); in Compile()
301 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index, in Compile()
303 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(), in Compile()
304 feature_index + 1); in Compile()
324 xla::Mul(grad_backprop, xla::Sub(activations, mean, {feature_index})); in Compile()
332 xla::Mul(grad_backprop, xla::Mul(scratch1, scale), {feature_index}); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_rewriter.cc | 88 HloInstruction* feature_index = in HandleBatchNormInference() local
90 LiteralUtil::CreateR0(batch_norm->feature_index()))); in HandleBatchNormInference()
100 operands.push_back(feature_index); in HandleBatchNormInference()
140 HloInstruction* feature_index = in HandleBatchNormTraining() local
142 LiteralUtil::CreateR0(batch_norm->feature_index()))); in HandleBatchNormTraining()
151 operands.push_back(feature_index); in HandleBatchNormTraining()
229 HloInstruction* feature_index = in HandleBatchNormGrad() local
231 LiteralUtil::CreateR0(batch_norm->feature_index()))); in HandleBatchNormGrad()
263 operands.push_back(feature_index); in HandleBatchNormGrad()
|
| D | cudnn_batchnorm_runner.cc | 71 int64 feature_index) { in MakeBatchNormDescriptors() argument
87 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index]; in MakeBatchNormDescriptors()
100 .set_feature_map_count(shape.dimensions(feature_index)) in MakeBatchNormDescriptors()
124 MakeBatchNormDescriptors(shape, config.feature_index); in AssignCommonParams()
214 int64 feature_index) { in GetCudnnBatchNormConfig() argument
222 config.feature_index = feature_index; in GetCudnnBatchNormConfig()
|
| D | gpu_layout_assignment_test.cc | 144 auto* feature_index = in TEST_F() local
150 {operand, scale, offset, mean, variance, epsilon, feature_index}, in TEST_F()
214 auto* feature_index = in TEST_F() local
219 batchnorm_shape, {operand, scale, offset, epsilon, feature_index}, in TEST_F()
292 auto* feature_index = in TEST_F() local
300 feature_index}, in TEST_F()
|
| D | cudnn_batchnorm_runner.h | 35 int64 feature_index; member
40 int64 feature_index);
|
| /external/libaom/libaom/av1/encoder/ |
| D | partition_strategy.c | 1025 int feature_index = 0; in av1_ml_prune_ab_partition() local
1026 features[feature_index++] = (float)part_ctx; in av1_ml_prune_ab_partition()
1027 features[feature_index++] = (float)var_ctx; in av1_ml_prune_ab_partition()
1051 features[feature_index++] = rd_ratio; in av1_ml_prune_ab_partition()
1053 assert(feature_index == 10); in av1_ml_prune_ab_partition()
1112 int feature_index = 0; in av1_ml_prune_4_partition() local
1113 features[feature_index++] = (float)part_ctx; in av1_ml_prune_4_partition()
1114 features[feature_index++] = (float)get_unsigned_bits(pb_source_variance); in av1_ml_prune_4_partition()
1139 features[feature_index++] = rd_ratio; in av1_ml_prune_4_partition()
1184 features[feature_index++] = var_ratio; in av1_ml_prune_4_partition()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | unfuse_batch_norm.mlir | 25 {epsilon = 1.001000e-05 : f32, feature_index = 1 : i64} :
44 {epsilon = 1.001000e-05 : f32, feature_index = 2 : i64} :
59 {epsilon = 1.0 : f32, feature_index = 1 : i64} :
74 {epsilon = 1.0 : f32, feature_index = 1 : i64} :
88 {epsilon = 0.00000001 : f32, feature_index = 1 : i64} :
130 {epsilon = 0.001 : f32, feature_index = 1 : i64} :
|
| D | lhlo_gpu_ops.mlir | 8 …, %grad_operand, %grad_scale, %grad_offset) {epsilon = 1.000000e-03 : f32, feature_index = 3 : i64}
17 …arg0, %arg1, %arg2, %arg3, %arg4, %arg_out) {epsilon = 1.000000e-03 : f32, feature_index = 3 : i64}
26 …1, %arg2, %output, %batch_mean, %batch_var) {epsilon = 1.000000e-03 : f32, feature_index = 3 : i64}
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_internal.cc | 170 const int64 feature_index = (*sparse_features.indices)(k); in ComputeWxAndWeightedExampleNorm() local
175 const float sparse_weight = sparse_weights.nominals(l, feature_index); in ComputeWxAndWeightedExampleNorm()
178 sparse_weights.deltas(l, feature_index) * num_loss_partitions; in ComputeWxAndWeightedExampleNorm()
431 const int64 feature_index = (*sparse_features->indices)(k); in CreateSparseFeatureRepresentation() local
432 if (!weights.SparseIndexValid(i, feature_index)) { in CreateSparseFeatureRepresentation()
518 const int64 feature_index = (*sparse_features.indices)(k); in ComputeSquaredNormPerExample() local
519 if (previous_indices.insert(feature_index).second == false) { in ComputeSquaredNormPerExample()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops.td | 59 I64Attr:$feature_index
73 I64Attr:$feature_index);
87 I64Attr:$feature_index
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1863 feature_index = 0
1870 activations[feature] = table_activations[:, feature_index, :]
1871 feature_index = feature_index + 1
1875 feature_index:(feature_index + seq_length), :])
1876 feature_index = feature_index + seq_length
|
| D | tpu_embedding_v2.py | 769 feature_index = table_to_position[feature.table]
774 activation[:, feature_index, :])
778 activation[:, feature_index:(
779 feature_index+feature.max_sequence_length), :])
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 540 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat() local
547 dim_sizes[feature_index] = C; in ShapeFromFormat()
|
| /external/fonttools/Lib/fontTools/feaLib/ |
| D | builder.py | 867 feature_index = feature_indices.get(feature_key)
868 if feature_index is None:
869 feature_index = len(table.FeatureList.FeatureRecord)
877 feature_indices[feature_key] = feature_index
878 scripts.setdefault(script, {}).setdefault(lang, []).append(feature_index)
880 required_feature_indices[(script, lang)] = feature_index
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 126 int64_t feature_dim = bn_op.feature_index(); in matchAndRewrite()
|