Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc73 HloInstruction* feature_index = in HandleBatchNormInference() local
109 HloInstruction* feature_index = in HandleBatchNormTraining() local
178 HloInstruction* feature_index = in HandleBatchNormGrad() local
Dcudnn_batchnorm_thunk.cc35 MakeDescriptors(const Shape& shape, int64 feature_index) { in MakeDescriptors()
82 const BufferAllocation::Slice& variance, float epsilon, int64 feature_index, in CudnnBatchNormForwardInferenceThunk()
138 float epsilon, int64 feature_index, in CudnnBatchNormForwardTrainingThunk()
219 int64 feature_index, const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk()
Dgpu_layout_assignment_test.cc142 auto* feature_index = in TEST_F() local
212 auto* feature_index = in TEST_F() local
290 auto* feature_index = in TEST_F() local
Dir_emitter_unnested.cc386 const HloInstruction* feature_index = custom_call->operand(6); in HandleCustomCall() local
410 const HloInstruction* feature_index = custom_call->operand(4); in HandleCustomCall() local
440 const HloInstruction* feature_index = custom_call->operand(6); in HandleCustomCall() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc53 int feature_index = in Compile() local
142 const int feature_index = in Compile() local
/external/harfbuzz_ng/src/
Dhb-ot-layout.cc461 unsigned int *feature_index) in hb_ot_layout_table_find_feature()
544 unsigned int *feature_index) in hb_ot_layout_language_get_required_feature_index()
564 unsigned int *feature_index, in hb_ot_layout_language_get_required_feature()
623 unsigned int *feature_index) in hb_ot_layout_language_find_feature()
651 unsigned int feature_index, in hb_ot_layout_feature_get_lookups()
761 unsigned int feature_index = l.get_feature_index (i); in langsys_collect_features() local
864 for (hb_codepoint_t feature_index = HB_SET_VALUE_INVALID; in hb_ot_layout_collect_lookups() local
924 unsigned int feature_index, in hb_ot_layout_feature_with_variations_get_lookups()
1194 unsigned int feature_index, in hb_ot_layout_feature_get_name_ids()
1267 unsigned int feature_index, in hb_ot_layout_feature_get_characters()
Dhb-ot-map.cc96 unsigned int feature_index, in add_lookups()
223 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()
176 unsigned int feature_index) in arabic_fallback_synthesize_lookup()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.cc46 for (int feature_index = 0; feature_index < table.num_features(); in AddDefaultEmbeddingOutputLayoutIfNeeded() 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
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc113 HloInstruction* operand, int64 feature_index, in DynamicElementCountPerFeature()
208 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormTraining() local
334 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormInference() local
470 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormGrad() local
Dshape_inference.cc1168 const Shape& offset_shape, int64 feature_index) { in InferBatchNormTrainingShape()
1274 const Shape& variance_shape, int64 feature_index) { in InferBatchNormInferenceShape()
1418 const Shape& output_grad_shape, int64 feature_index) { in InferBatchNormGradShape()
Dhlo.proto121 int64 feature_index = 25; field
Dhlo_instruction.cc1066 int64 feature_index) { in CreateBatchNormTraining()
1075 float epsilon, int64 feature_index) { in CreateBatchNormInference()
1085 int64 feature_index) { in CreateBatchNormGrad()
3214 int64 HloInstruction::feature_index() const { in feature_index() function in xla::HloInstruction
Dhlo_instructions.h30 int64 feature_index() const { return feature_index_; } in feature_index() function
Dhlo_parser.cc1317 optional<int64> feature_index; in ParseInstructionRhs() local
1333 optional<int64> feature_index; in ParseInstructionRhs() local
1350 optional<int64> feature_index; in ParseInstructionRhs() local
Dhlo_instructions.cc75 HloInstruction* scale, float epsilon, int64 feature_index) in HloBatchNormInstruction()
107 HloInstruction* offset, float epsilon, int64 feature_index) in HloBatchNormTrainingInstruction()
126 float epsilon, int64 feature_index) in HloBatchNormInferenceInstruction()
147 float epsilon, int64 feature_index) in HloBatchNormGradInstruction()
/external/tensorflow/tensorflow/contrib/libsvm/kernels/
Ddecode_libsvm_op.cc70 int64 feature_index; in Compute() local
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc170 const int64 feature_index = (*sparse_features.indices)(k); in ComputeWxAndWeightedExampleNorm() local
431 const int64 feature_index = (*sparse_features->indices)(k); in CreateSparseFeatureRepresentation() local
518 const int64 feature_index = (*sparse_features.indices)(k); in ComputeSquaredNormPerExample() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc373 int64 feature_index; member
409 float random_value_mean, float random_value_var) { in BuildBatchNormTestParams()
457 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
558 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
670 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.h229 const int64 feature_index = indices(index, 0); in GetNumSparseFeatures() local
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc87 int32 GetFeatureDimension(const int32 feature_index, const int64 instance, in GetFeatureDimension()
110 for (int32 feature_index = 0; feature_index < values_list.size(); in QuantizeFeatures() local
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h527 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2047 int64 feature_index) { in BatchNormTraining()
2071 int64 feature_index) { in BatchNormInference()
2097 int64 feature_index) { in BatchNormGrad()
3488 int64 feature_index) { in BatchNormTraining()
3496 int64 feature_index) { in BatchNormInference()
3504 int64 feature_index) { in BatchNormGrad()
/external/libaom/libaom/av1/encoder/
Dencodeframe.c2643 int feature_index = 0; in ml_prune_2pass_split_partition() local
2795 int feature_index = 0; in ml_prune_ab_partition() local
2882 int feature_index = 0; in ml_prune_4_partition() local
3058 int feature_index = 0; in ml_predict_breakout() local

12