Home
last modified time | relevance | path

Searched defs:feature_index (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc88 HloInstruction* feature_index = in HandleBatchNormInference() local
140 HloInstruction* feature_index = in HandleBatchNormTraining() local
229 HloInstruction* feature_index = in HandleBatchNormGrad() local
Dcudnn_batchnorm_runner.h35 int64 feature_index; member
Dgpu_layout_assignment_test.cc143 auto* feature_index = in TEST_F() local
212 auto* feature_index = in TEST_F() local
289 auto* feature_index = in TEST_F() local
Dcudnn_batchnorm_runner.cc71 int64_t feature_index) { in MakeBatchNormDescriptors()
215 int64_t feature_index) { in GetCudnnBatchNormConfig()
/external/harfbuzz_ng/src/
Dhb-ot-layout.cc628 unsigned int *feature_index /* OUT */) in hb_ot_layout_table_find_feature()
778 unsigned int *feature_index /* OUT */) in hb_ot_layout_language_get_required_feature_index()
810 unsigned int *feature_index /* OUT */, in hb_ot_layout_language_get_required_feature()
920 unsigned int *feature_index /* OUT */) in hb_ot_layout_language_find_feature()
962 unsigned int feature_index, in hb_ot_layout_feature_get_lookups()
1114 unsigned int feature_index = l.get_feature_index (i); in langsys_collect_features() local
1236 for (hb_codepoint_t feature_index = HB_SET_VALUE_INVALID; in hb_ot_layout_collect_lookups() local
1345 unsigned int feature_index, in hb_ot_layout_feature_with_variations_get_lookups()
1706 unsigned int feature_index, in hb_ot_layout_feature_get_name_ids()
1775 unsigned int feature_index, in hb_ot_layout_feature_get_characters()
Dhb-ot-map.cc115 unsigned int feature_index, in add_lookups()
243 unsigned int feature_index[2]; in compile() local
Dhb-ot-shape-complex-arabic-fallback.hh50 unsigned int feature_index) in arabic_fallback_synthesize_lookup_single()
179 unsigned int feature_index) in arabic_fallback_synthesize_lookup()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc105 HloInstruction* operand, int64_t feature_index, in DynamicElementCountPerFeature()
175 int64_t feature_index = batch_norm->feature_index(); in HandleBatchNormTraining() local
301 int64_t feature_index = batch_norm->feature_index(); in HandleBatchNormInference() local
419 int64_t feature_index = batch_norm->feature_index(); in HandleBatchNormGrad() local
Dshape_inference.cc1201 const Shape& offset_shape, int64_t feature_index) { in InferBatchNormTrainingShape()
1307 const Shape& variance_shape, int64_t feature_index) { in InferBatchNormInferenceShape()
1451 const Shape& output_grad_shape, int64_t feature_index) { in InferBatchNormGradShape()
Dhlo.proto162 int64 feature_index = 25; field
Dhlo_instruction.cc1504 int64_t feature_index) { in CreateBatchNormTraining()
1513 float epsilon, int64_t feature_index) { in CreateBatchNormInference()
1523 int64_t feature_index) { in CreateBatchNormGrad()
4191 int64 HloInstruction::feature_index() const { in feature_index() function in xla::HloInstruction
Dhlo_instructions.cc81 HloInstruction* scale, float epsilon, int64_t feature_index) in HloBatchNormInstruction()
113 HloInstruction* offset, float epsilon, int64_t feature_index) in HloBatchNormTrainingInstruction()
132 float epsilon, int64_t feature_index) in HloBatchNormInferenceInstruction()
153 float epsilon, int64_t feature_index) in HloBatchNormGradInstruction()
Dhlo_instructions.h33 int64 feature_index() const { return feature_index_; } in feature_index() function
Dhlo_parser.cc2114 optional<int64> feature_index; in ParseInstructionRhs() local
2139 optional<int64> feature_index; in ParseInstructionRhs() local
2166 optional<int64> feature_index; in ParseInstructionRhs() local
/external/harfbuzz_ng/test/api/
Dtest-ot-name.c36 unsigned int feature_index; in test_ot_layout_feature_get_name_ids_and_characters() local
Dtest-ot-face.c114 unsigned feature_index = 0; in test_font() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc85 int feature_index = in CompileImpl() local
269 const int feature_index = in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc184 const int64_t feature_index = (*sparse_features.indices)(k); in ComputeWxAndWeightedExampleNorm() local
472 const int64_t feature_index = (*sparse_features->indices)(k); in CreateSparseFeatureRepresentation() local
559 const int64_t feature_index = (*sparse_features.indices)(k); in ComputeSquaredNormPerExample() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc471 int64 feature_index; member
507 float random_value_mean, float random_value_var) { in BuildBatchNormTestParams()
555 const int64_t feature_index = GetParam().feature_index; in XLA_TEST_P() local
656 const int64_t feature_index = GetParam().feature_index; in XLA_TEST_P() local
768 const int64_t feature_index = GetParam().feature_index; in XLA_TEST_P() local
/external/libaom/libaom/av1/encoder/
Dpartition_strategy.c1212 int feature_index = 0; in av1_ml_prune_ab_partition() local
1323 int feature_index = 0; in av1_ml_prune_4_partition() local
1485 int feature_index = 0; in av1_ml_predict_breakout() local
1851 int feature_index = 0; in prepare_features_after_part_ab() local
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h540 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat() local
Dexample_proto_fast_parsing.cc1285 size_t feature_index = 0; in FastParseExample() local
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc518 size_t feature_index = 0; in FastParseExampleLite() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2670 float epsilon, int64_t feature_index) { in BatchNormTraining()
2693 int64_t feature_index) { in BatchNormInference()
2718 float epsilon, int64_t feature_index) { in BatchNormGrad()
4803 int64_t feature_index) { in BatchNormTraining()
4811 int64_t feature_index) { in BatchNormInference()
4819 int64_t feature_index) { in BatchNormGrad()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc969 const HloInstruction* feature_index = in EmitDnnBatchNorm() local

12