Home
last modified time | relevance | path

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

/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.cc70 int64 feature_index) { in MakeBatchNormDescriptors()
116 int64 feature_index) { in AssignCommonParams()
222 se::DeviceMemory<float> variance, float epsilon, int64 feature_index, in RunCudnnBatchNormForwardInference()
252 se::DeviceMemory<float> offset, float epsilon, int64 feature_index, in RunCudnnBatchNormForwardTraining()
285 float epsilon, int64 feature_index, se::Stream* stream) { in RunCudnnBatchNormBackward()
Dcudnn_batchnorm_thunk.cc98 const BufferAllocation::Slice& variance, float epsilon, int64 feature_index, in CudnnBatchNormForwardInferenceThunk()
144 float epsilon, int64 feature_index, in CudnnBatchNormForwardTrainingThunk()
209 int64 feature_index, const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk()
Dgpu_layout_assignment_test.cc144 auto* feature_index = in TEST_F() local
214 auto* feature_index = in TEST_F() local
292 auto* feature_index = in TEST_F() local
Dthunk_emitter.cc146 const HloInstruction* feature_index = custom_call->operand(6); in HandleCustomCall() local
170 const HloInstruction* feature_index = custom_call->operand(4); in HandleCustomCall() local
199 const HloInstruction* feature_index = custom_call->operand(6); in HandleCustomCall() local
/external/harfbuzz_ng/src/
Dhb-ot-map.cc100 unsigned int feature_index, in add_lookups()
228 unsigned int feature_index[2]; in compile() local
Dhb-ot-layout.cc645 unsigned int *feature_index /* OUT */) in hb_ot_layout_table_find_feature()
788 unsigned int *feature_index) in hb_ot_layout_language_get_required_feature_index()
820 unsigned int *feature_index, in hb_ot_layout_language_get_required_feature()
927 unsigned int *feature_index /* OUT */) in hb_ot_layout_language_find_feature()
967 unsigned int feature_index, in hb_ot_layout_feature_get_lookups()
1083 unsigned int feature_index = l.get_feature_index (i); in langsys_collect_features() local
1212 for (hb_codepoint_t feature_index = HB_SET_VALUE_INVALID; in hb_ot_layout_collect_lookups() local
1315 unsigned int feature_index, in hb_ot_layout_feature_with_variations_get_lookups()
1664 unsigned int feature_index, in hb_ot_layout_feature_get_name_ids()
1739 unsigned int feature_index, in hb_ot_layout_feature_get_characters()
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/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.cc105 HloInstruction* operand, int64 feature_index, in DynamicElementCountPerFeature()
175 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormTraining() local
301 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormInference() local
419 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormGrad() local
Dshape_inference.cc1192 const Shape& offset_shape, int64 feature_index) { in InferBatchNormTrainingShape()
1298 const Shape& variance_shape, int64 feature_index) { in InferBatchNormInferenceShape()
1442 const Shape& output_grad_shape, int64 feature_index) { in InferBatchNormGradShape()
Dhlo.proto127 int64 feature_index = 25; field
Dhlo_instruction.cc1161 int64 feature_index) { in CreateBatchNormTraining()
1170 float epsilon, int64 feature_index) { in CreateBatchNormInference()
1180 int64 feature_index) { in CreateBatchNormGrad()
3544 int64 HloInstruction::feature_index() const { in feature_index() function in xla::HloInstruction
Dhlo_parser.cc1379 optional<int64> feature_index; in ParseInstructionRhs() local
1395 optional<int64> feature_index; in ParseInstructionRhs() local
1412 optional<int64> feature_index; in ParseInstructionRhs() local
Dhlo_instructions.h30 int64 feature_index() const { return feature_index_; } in feature_index() function
Dhlo_instructions.cc76 HloInstruction* scale, float epsilon, int64 feature_index) in HloBatchNormInstruction()
108 HloInstruction* offset, float epsilon, int64 feature_index) in HloBatchNormTrainingInstruction()
127 float epsilon, int64 feature_index) in HloBatchNormInferenceInstruction()
148 float epsilon, int64 feature_index) in HloBatchNormGradInstruction()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc83 int feature_index = in CompileImpl() local
246 const int feature_index = in Compile() 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.cc471 int64 feature_index; member
507 float random_value_mean, float random_value_var) { in BuildBatchNormTestParams()
555 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
656 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
768 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
/external/libaom/libaom/av1/encoder/
Dpartition_strategy.c1025 int feature_index = 0; in av1_ml_prune_ab_partition() local
1112 int feature_index = 0; in av1_ml_prune_4_partition() local
1260 int feature_index = 0; in av1_ml_predict_breakout() local
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h539 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat() local
Dexample_proto_fast_parsing.cc1264 size_t feature_index = 0; in FastParseExample() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2011 float epsilon, int64 feature_index) { in BatchNormTraining()
2034 int64 feature_index) { in BatchNormInference()
2059 float epsilon, int64 feature_index) { in BatchNormGrad()
3556 int64 feature_index) { in BatchNormTraining()
3564 int64 feature_index) { in BatchNormInference()
3572 int64 feature_index) { in BatchNormGrad()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_encodeframe.c3339 int feature_index = 0; in ml_predict_breakout() local
3428 int feature_index = 0; in ml_prune_rect_partition() local