Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc364 int64 feature_index; member
372 os << "feature_index=" << p.feature_index << ", "; in operator <<()
399 auto add_testcase = [&](std::vector<int64> bounds, int64 feature_index, in BuildBatchNormTestParams()
401 BatchNormTestParam p{bounds, feature_index, random_value_mean, in BuildBatchNormTestParams()
448 const int64 feature_index = GetParam().feature_index; in XLA_TEST_P() local
450 Product(bounds) / bounds[feature_index]; in XLA_TEST_P()
451 const int64 feature_bound = bounds[feature_index]; in XLA_TEST_P()
459 if (i != feature_index) { in XLA_TEST_P()
494 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
495 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc165 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormTraining() local
166 const int64 feature_count = operand_shape.dimensions(feature_index); in HandleBatchNormTraining()
190 if (i != feature_index) { in HandleBatchNormTraining()
196 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index})); in HandleBatchNormTraining()
199 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index})); in HandleBatchNormTraining()
236 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index})); in HandleBatchNormTraining()
251 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index})); in HandleBatchNormTraining()
312 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormInference() local
329 if (i != feature_index) { in HandleBatchNormInference()
343 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index})); in HandleBatchNormInference()
[all …]
Dshape_inference.h94 int64 feature_index);
101 const Shape& variance_shape, int64 feature_index);
109 int64 feature_index);
Dshape_inference.cc1181 const Shape& offset_shape, int64 feature_index) { in InferBatchNormTrainingShape() argument
1196 if (feature_index >= ShapeUtil::Rank(operand_shape)) { in InferBatchNormTrainingShape()
1201 feature_index, ShapeUtil::Rank(operand_shape)); in InferBatchNormTrainingShape()
1204 if (feature_index < 0) { in InferBatchNormTrainingShape()
1208 feature_index); in InferBatchNormTrainingShape()
1259 const int64 feature_count = operand_shape.dimensions(feature_index); in InferBatchNormTrainingShape()
1287 const Shape& variance_shape, int64 feature_index) { in InferBatchNormInferenceShape() argument
1306 if (feature_index >= ShapeUtil::Rank(operand_shape)) { in InferBatchNormInferenceShape()
1311 feature_index, ShapeUtil::Rank(operand_shape)); in InferBatchNormInferenceShape()
1314 if (feature_index < 0) { in InferBatchNormInferenceShape()
[all …]
Dhlo_instruction.h383 HloInstruction* offset, float epsilon, int64 feature_index);
389 float epsilon, int64 feature_index);
395 HloInstruction* grad_output, float epsilon, int64 feature_index);
832 int64 feature_index() const { return feature_index_; } in feature_index() function
Dhlo_instruction.cc143 instruction->feature_index_ = proto.feature_index(); in CreateFromProto()
601 int64 feature_index) { in CreateBatchNormTraining() argument
608 instruction->feature_index_ = feature_index; in CreateBatchNormTraining()
616 float epsilon, int64 feature_index) { in CreateBatchNormInference() argument
625 instruction->feature_index_ = feature_index; in CreateBatchNormInference()
634 int64 feature_index) { in CreateBatchNormGrad() argument
643 instruction->feature_index_ = feature_index; in CreateBatchNormGrad()
1386 new_operands[2], epsilon(), feature_index()); in CloneWithNewOperands()
1392 new_operands[3], new_operands[4], epsilon(), feature_index()); in CloneWithNewOperands()
1406 new_operands[4], epsilon(), feature_index()); in CloneWithNewOperands()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc56 int feature_index = in Compile() local
66 input, ctx->Input(1), ctx->Input(2), epsilon_, feature_index); in Compile()
84 epsilon_, feature_index); in Compile()
130 int feature_index = in Compile() local
152 epsilon_, feature_index); in Compile()
160 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index, in Compile()
162 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(), in Compile()
163 feature_index + 1); in Compile()
179 b->Mul(grad_backprop, b->Sub(activations, mean, {feature_index})), in Compile()
184 b->Mul(grad_backprop, b->Mul(scratch1, scale), {feature_index}); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc71 HloInstruction* feature_index = in HandleBatchNormInference() local
73 Literal::CreateR0(batch_norm->feature_index()))); in HandleBatchNormInference()
78 operands.push_back(feature_index); in HandleBatchNormInference()
106 HloInstruction* feature_index = in HandleBatchNormTraining() local
108 Literal::CreateR0(batch_norm->feature_index()))); in HandleBatchNormTraining()
113 operands.push_back(feature_index); in HandleBatchNormTraining()
169 HloInstruction* feature_index = in HandleBatchNormGrad() local
171 Literal::CreateR0(batch_norm->feature_index()))); in HandleBatchNormGrad()
189 operands.push_back(feature_index); in HandleBatchNormGrad()
Dcudnn_batchnorm_thunk.cc36 MakeDescriptors(const Shape& shape, int64 feature_index) { in MakeDescriptors() argument
52 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index]; in MakeDescriptors()
65 .set_feature_map_count(shape.dimensions(feature_index)) in MakeDescriptors()
83 const BufferAllocation::Slice& variance, float epsilon, int64 feature_index, in CudnnBatchNormForwardInferenceThunk() argument
92 feature_index_(feature_index), in CudnnBatchNormForwardInferenceThunk()
136 float epsilon, int64 feature_index, in CudnnBatchNormForwardTrainingThunk() argument
146 feature_index_(feature_index), in CudnnBatchNormForwardTrainingThunk()
215 int64 feature_index, const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk() argument
226 feature_index_(feature_index), in CudnnBatchNormBackwardThunk()
Dgpu_layout_assignment_test.cc135 auto* feature_index = in TEST_F() local
141 {operand, scale, offset, mean, variance, epsilon, feature_index}, in TEST_F()
203 auto* feature_index = in TEST_F() local
208 batchnorm_shape, {operand, scale, offset, epsilon, feature_index}, in TEST_F()
279 auto* feature_index = in TEST_F() local
287 feature_index}, in TEST_F()
Dcudnn_batchnorm_thunk.h53 float epsilon, int64 feature_index,
81 const BufferAllocation::Slice& offset, float epsilon, int64 feature_index,
114 float epsilon, int64 feature_index,
Dir_emitter_unnested.cc330 const HloInstruction* feature_index = custom_call->operand(6); in HandleCustomCall() local
331 CHECK(feature_index->IsConstant()); in HandleCustomCall()
332 int64 feature_index_value = feature_index->literal().Get<int64>({}); in HandleCustomCall()
354 const HloInstruction* feature_index = custom_call->operand(4); in HandleCustomCall() local
355 CHECK(feature_index->IsConstant()); in HandleCustomCall()
356 int64 feature_index_value = feature_index->literal().Get<int64>({}); in HandleCustomCall()
384 const HloInstruction* feature_index = custom_call->operand(6); in HandleCustomCall() local
385 CHECK(feature_index->IsConstant()); in HandleCustomCall()
386 int64 feature_index_value = feature_index->literal().Get<int64>({}); in HandleCustomCall()
/external/harfbuzz_ng/src/
Dhb-ot-layout.cc416 unsigned int *feature_index) in hb_ot_layout_table_find_feature() argument
425 if (feature_index) *feature_index = i; in hb_ot_layout_table_find_feature()
430 if (feature_index) *feature_index = HB_OT_LAYOUT_NO_FEATURE_INDEX; in hb_ot_layout_table_find_feature()
474 unsigned int *feature_index) in hb_ot_layout_language_get_required_feature_index() argument
480 feature_index, in hb_ot_layout_language_get_required_feature_index()
494 unsigned int *feature_index, in hb_ot_layout_language_get_required_feature() argument
501 if (feature_index) *feature_index = index; in hb_ot_layout_language_get_required_feature()
553 unsigned int *feature_index) in hb_ot_layout_language_find_feature() argument
564 if (feature_index) *feature_index = f_index; in hb_ot_layout_language_find_feature()
569 if (feature_index) *feature_index = HB_OT_LAYOUT_NO_FEATURE_INDEX; in hb_ot_layout_language_find_feature()
[all …]
Dhb-ot-layout.h148 unsigned int *feature_index);
155 unsigned int *feature_index,
182 unsigned int *feature_index);
187 unsigned int feature_index,
252 unsigned int feature_index,
Dhb-ot-map.cc85 unsigned int feature_index, in add_lookups() argument
102 feature_index, in add_lookups()
217 unsigned int feature_index[2]; in compile() local
228 &feature_index[table_index]); in compile()
237 &feature_index[table_index]); in compile()
249 map->index[0] = feature_index[0]; in compile()
250 map->index[1] = feature_index[1]; in compile()
Dhb-ot-shape-complex-arabic-fallback.hh50 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()
186 unsigned int feature_index) in arabic_fallback_synthesize_lookup() argument
188 if (feature_index < 4) in arabic_fallback_synthesize_lookup()
189 return arabic_fallback_synthesize_lookup_single (plan, font, feature_index); in arabic_fallback_synthesize_lookup()
Dhb-ot-layout-common-private.hh1503 inline const Feature *find_substitute (unsigned int feature_index) const in find_substitute()
1509 if (record.featureIndex == feature_index) in find_substitute()
1573 unsigned int feature_index) const in find_substitute()
1576 return (this+record.substitutions).find_substitute (feature_index); in find_substitute()
/external/tensorflow/tensorflow/contrib/libsvm/kernels/
Ddecode_libsvm_op.cc70 int64 feature_index; in Compute() local
72 ctx, strings::safe_strto64(piece.substr(0, p), &feature_index), in Compute()
74 OP_REQUIRES(ctx, (feature_index >= 0), in Compute()
76 "Feature index should be >= 0, got ", feature_index)); in Compute()
87 out_indices.emplace_back(std::pair<int64, int64>(i, feature_index)); in Compute()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.h229 const int64 feature_index = indices(index, 0); in GetNumSparseFeatures() local
230 if (feature_index == input_index) { in GetNumSparseFeatures()
233 } else if (feature_index < input_index) { in GetNumSparseFeatures()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc87 int32 GetFeatureDimension(const int32 feature_index, const int64 instance, in GetFeatureDimension() argument
91 return (*indices_list)[feature_index].matrix<int64>()(instance, 1); in GetFeatureDimension()
110 for (int32 feature_index = 0; feature_index < values_list.size(); in QuantizeFeatures() local
111 ++feature_index) { in QuantizeFeatures()
112 const Tensor& values_tensor = values_list[feature_index]; in QuantizeFeatures()
118 context, output_list.allocate(feature_index, in QuantizeFeatures()
124 GetBuckets(feature_index, buckets_list); in QuantizeFeatures()
139 GetFeatureDimension(feature_index, instance, indices_list); in QuantizeFeatures()
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc164 const int64 feature_index = (*sparse_features.indices)(k); in ComputeWxAndWeightedExampleNorm() local
169 const float sparse_weight = sparse_weights.nominals(l, feature_index); in ComputeWxAndWeightedExampleNorm()
172 sparse_weights.deltas(l, feature_index) * num_loss_partitions; in ComputeWxAndWeightedExampleNorm()
417 const int64 feature_index = (*sparse_features->indices)(k); in CreateSparseFeatureRepresentation() local
418 if (!weights.SparseIndexValid(i, feature_index)) { in CreateSparseFeatureRepresentation()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h420 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat() local
424 dim_sizes[feature_index] = C / 4; in ShapeFromFormat()
427 dim_sizes[feature_index] = C; in ShapeFromFormat()
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc849 optional<int64> feature_index; in ParseInstruction() local
851 &feature_index}; in ParseInstruction()
859 /*offset=*/operands[2], *epsilon, *feature_index)); in ParseInstruction()
865 optional<int64> feature_index; in ParseInstruction() local
867 &feature_index}; in ParseInstruction()
876 /*variance=*/operands[4], *epsilon, *feature_index)); in ParseInstruction()
882 optional<int64> feature_index; in ParseInstruction() local
884 &feature_index}; in ParseInstruction()
892 /*grad_output=*/operands[4], *epsilon, *feature_index)); in ParseInstruction()
/external/tensorflow/tensorflow/compiler/xla/client/
Dcomputation_builder.cc1354 const ComputationDataHandle& offset, float epsilon, int64 feature_index) { in BatchNormTraining() argument
1362 request->set_feature_index(feature_index); in BatchNormTraining()
1369 const ComputationDataHandle& variance, float epsilon, int64 feature_index) { in BatchNormInference() argument
1379 request->set_feature_index(feature_index); in BatchNormInference()
1387 int64 feature_index) { in BatchNormGrad() argument
1396 request->set_feature_index(feature_index); in BatchNormGrad()
Dcomputation_builder.h753 float epsilon, int64 feature_index);
769 int64 feature_index);
785 float epsilon, int64 feature_index);

12