Searched defs:feature_index (Results 1 - 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_batchnorm_rewriter.cc71 HloInstruction* feature_index = local
73 Literal::CreateR0(batch_norm->feature_index())));
78 operands.push_back(feature_index);
106 HloInstruction* feature_index = local
108 Literal::CreateR0(batch_norm->feature_index())));
113 operands.push_back(feature_index);
169 HloInstruction* feature_index = local
171 Literal::CreateR0(batch_norm->feature_index())));
189 operands.push_back(feature_index);
H A Dgpu_layout_assignment_test.cc135 auto* feature_index = local
141 {operand, scale, offset, mean, variance, epsilon, feature_index},
203 auto* feature_index = local
208 batchnorm_shape, {operand, scale, offset, epsilon, feature_index},
279 auto* feature_index = local
287 feature_index},
H A Dcudnn_batchnorm_thunk.cc36 MakeDescriptors(const Shape& shape, int64 feature_index) { argument
52 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index];
65 .set_feature_map_count(shape.dimensions(feature_index))
83 const BufferAllocation::Slice& variance, float epsilon, int64 feature_index,
92 feature_index_(feature_index),
136 float epsilon, int64 feature_index,
146 feature_index_(feature_index),
215 int64 feature_index, const BufferAllocation::Slice& output_grad_data,
226 feature_index_(feature_index),
79 CudnnBatchNormForwardInferenceThunk( const BufferAllocation::Slice& operand, const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset, const BufferAllocation::Slice& mean, const BufferAllocation::Slice& variance, float epsilon, int64 feature_index, const BufferAllocation::Slice& output, const HloInstruction* hlo) argument
133 CudnnBatchNormForwardTrainingThunk( const BufferAllocation::Slice& operand, const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset, float epsilon, int64 feature_index, const BufferAllocation::Slice& output_data, const BufferAllocation::Slice& output_mean, const BufferAllocation::Slice& output_inv_stddev, const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) argument
210 CudnnBatchNormBackwardThunk( const BufferAllocation::Slice& operand, const BufferAllocation::Slice& scale, const BufferAllocation::Slice& mean, const BufferAllocation::Slice& inv_stddev, const BufferAllocation::Slice& grad_output, float epsilon, int64 feature_index, const BufferAllocation::Slice& output_grad_data, const BufferAllocation::Slice& output_grad_scale, const BufferAllocation::Slice& output_grad_offset, const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) argument
H A Dir_emitter_unnested.cc323 // feature_index operands be constants.
330 const HloInstruction* feature_index = custom_call->operand(6); local
331 CHECK(feature_index->IsConstant());
332 int64 feature_index_value = feature_index->literal().Get<int64>({});
342 /*feature_index=*/feature_index_value,
354 const HloInstruction* feature_index = custom_call->operand(4); local
355 CHECK(feature_index->IsConstant());
356 int64 feature_index_value = feature_index->literal().Get<int64>({});
370 /*feature_index=*/feature_index_value,
384 const HloInstruction* feature_index local
[all...]
/external/tensorflow/tensorflow/contrib/libsvm/kernels/
H A Ddecode_libsvm_op.cc70 int64 feature_index; variable
72 ctx, strings::safe_strto64(piece.substr(0, p), &feature_index),
74 OP_REQUIRES(ctx, (feature_index >= 0),
76 "Feature index should be >= 0, got ", feature_index));
87 out_indices.emplace_back(std::pair<int64, int64>(i, feature_index));
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc56 int feature_index = variable
66 input, ctx->Input(1), ctx->Input(2), epsilon_, feature_index);
84 epsilon_, feature_index);
130 int feature_index = variable
152 epsilon_, feature_index);
160 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index,
162 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(),
163 feature_index + 1);
179 b->Mul(grad_backprop, b->Sub(activations, mean, {feature_index})),
184 b->Mul(grad_backprop, b->Mul(scratch1, scale), {feature_index});
[all...]
/external/harfbuzz_ng/src/
H A Dhb-ot-map.cc85 unsigned int feature_index,
102 feature_index,
217 unsigned int feature_index[2]; local
228 &feature_index[table_index]);
237 &feature_index[table_index]);
249 map->index[0] = feature_index[0];
250 map->index[1] = feature_index[1];
82 add_lookups(hb_ot_map_t &m, hb_face_t *face, unsigned int table_index, unsigned int feature_index, unsigned int variations_index, hb_mask_t mask, bool auto_zwnj, bool auto_zwj) argument
H A Dhb-ot-shape-complex-arabic-fallback.hh50 unsigned int feature_index)
59 hb_codepoint_t s = shaping_table[u - SHAPING_TABLE_FIRST][feature_index];
186 unsigned int feature_index)
188 if (feature_index < 4)
189 return arabic_fallback_synthesize_lookup_single (plan, font, feature_index);
48 arabic_fallback_synthesize_lookup_single(const hb_ot_shape_plan_t *plan HB_UNUSED, hb_font_t *font, unsigned int feature_index) argument
184 arabic_fallback_synthesize_lookup(const hb_ot_shape_plan_t *plan, hb_font_t *font, unsigned int feature_index) argument
H A Dhb-ot-layout.cc416 unsigned int *feature_index)
425 if (feature_index) *feature_index = i;
430 if (feature_index) *feature_index = HB_OT_LAYOUT_NO_FEATURE_INDEX;
474 unsigned int *feature_index)
480 feature_index,
494 unsigned int *feature_index,
501 if (feature_index) *feature_index
413 hb_ot_layout_table_find_feature(hb_face_t *face, hb_tag_t table_tag, hb_tag_t feature_tag, unsigned int *feature_index) argument
470 hb_ot_layout_language_get_required_feature_index(hb_face_t *face, hb_tag_t table_tag, unsigned int script_index, unsigned int language_index, unsigned int *feature_index) argument
490 hb_ot_layout_language_get_required_feature(hb_face_t *face, hb_tag_t table_tag, unsigned int script_index, unsigned int language_index, unsigned int *feature_index, hb_tag_t *feature_tag) argument
548 hb_ot_layout_language_find_feature(hb_face_t *face, hb_tag_t table_tag, unsigned int script_index, unsigned int language_index, hb_tag_t feature_tag, unsigned int *feature_index) argument
579 hb_ot_layout_feature_get_lookups(hb_face_t *face, hb_tag_t table_tag, unsigned int feature_index, unsigned int start_offset, unsigned int *lookup_count , unsigned int *lookup_indexes ) argument
620 _hb_ot_layout_collect_lookups_lookups(hb_face_t *face, hb_tag_t table_tag, unsigned int feature_index, hb_set_t *lookup_indexes ) argument
693 unsigned int feature_index; local
859 hb_ot_layout_feature_with_variations_get_lookups(hb_face_t *face, hb_tag_t table_tag, unsigned int feature_index, unsigned int variations_index, unsigned int start_offset, unsigned int *lookup_count , unsigned int *lookup_indexes ) argument
[all...]
H A Dhb-ot-layout-gsubgpos-private.hh2326 inline const Feature& get_feature_variation (unsigned int feature_index, argument
2333 feature_index);
2337 return get_feature (feature_index);
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
H A Dtree_utils.h229 const int64 feature_index = indices(index, 0); local
230 if (feature_index == input_index) {
233 } else if (feature_index < input_index) {
/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h420 int feature_index = GetTensorFeatureDimIndex(dims, format); local
424 dim_sizes[feature_index] = C / 4;
427 dim_sizes[feature_index] = C;
/external/tensorflow/tensorflow/core/kernels/
H A Dsdca_internal.cc164 const int64 feature_index = (*sparse_features.indices)(k); local
169 const float sparse_weight = sparse_weights.nominals(l, feature_index);
172 sparse_weights.deltas(l, feature_index) * num_loss_partitions;
417 const int64 feature_index = (*sparse_features->indices)(k);
418 if (!weights.SparseIndexValid(i, feature_index)) {
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbatchnorm_expander.cc165 int64 feature_index = batch_norm->feature_index(); local
166 const int64 feature_count = operand_shape.dimensions(feature_index);
190 if (i != feature_index) {
196 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index}));
199 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index}));
236 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index}));
251 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index}));
312 int64 feature_index = batch_norm->feature_index(); local
436 int64 feature_index = batch_norm->feature_index(); local
[all...]
H A 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);
827 // Returns feature_index field associated with the instruction. The index
832 int64 feature_index() const { return feature_index_; } function in class:xla::HloInstruction
H A Dshape_inference.cc1181 const Shape& offset_shape, int64 feature_index) {
1196 if (feature_index >= ShapeUtil::Rank(operand_shape)) {
1198 "Expected feature_index of batch-norm-training to be "
1200 "got feature_index %lld, and rank %lld",
1201 feature_index, ShapeUtil::Rank(operand_shape));
1204 if (feature_index < 0) {
1206 "Expected feature_index of batch-norm-training to "
1208 feature_index);
1259 const int64 feature_count = operand_shape.dimensions(feature_index);
1287 const Shape& variance_shape, int64 feature_index) {
1179 InferBatchNormTrainingShape( const Shape& operand_shape, const Shape& scale_shape, const Shape& offset_shape, int64 feature_index) argument
1284 InferBatchNormInferenceShape( const Shape& operand_shape, const Shape& scale_shape, const Shape& offset_shape, const Shape& mean_shape, const Shape& variance_shape, int64 feature_index) argument
1432 InferBatchNormGradShape( const Shape& operand_shape, const Shape& scale_shape, const Shape& mean_shape, const Shape& var_shape, const Shape& output_grad_shape, int64 feature_index) argument
[all...]
H A Dhlo_instruction.cc143 instruction->feature_index_ = proto.feature_index();
601 int64 feature_index) {
608 instruction->feature_index_ = feature_index;
616 float epsilon, int64 feature_index) {
625 instruction->feature_index_ = feature_index;
634 int64 feature_index) {
643 instruction->feature_index_ = feature_index;
1386 new_operands[2], epsilon(), feature_index());
1392 new_operands[3], new_operands[4], epsilon(), feature_index());
1406 new_operands[4], epsilon(), feature_index());
597 CreateBatchNormTraining(const Shape& shape, HloInstruction* operand, HloInstruction* scale, HloInstruction* offset, float epsilon, int64 feature_index) argument
613 CreateBatchNormInference( const Shape& shape, HloInstruction* operand, HloInstruction* scale, HloInstruction* offset, HloInstruction* mean, HloInstruction* variance, float epsilon, int64 feature_index) argument
630 CreateBatchNormGrad(const Shape& shape, HloInstruction* operand, HloInstruction* scale, HloInstruction* mean, HloInstruction* variance, HloInstruction* grad_output, float epsilon, int64 feature_index) argument
[all...]
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc364 int64 feature_index; member in struct:xla::__anon25717::BatchNormTestParam
372 os << "feature_index=" << p.feature_index << ", ";
399 auto add_testcase = [&](std::vector<int64> bounds, int64 feature_index,
401 BatchNormTestParam p{bounds, feature_index, random_value_mean,
448 const int64 feature_index = GetParam().feature_index; local
450 Product(bounds) / bounds[feature_index];
451 const int64 feature_bound = bounds[feature_index];
459 if (i != feature_index) {
548 const int64 feature_index = GetParam().feature_index; local
660 const int64 feature_index = GetParam().feature_index; local
[all...]
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dquantile_ops.cc87 int32 GetFeatureDimension(const int32 feature_index, const int64 instance, argument
91 return (*indices_list)[feature_index].matrix<int64>()(instance, 1);
110 for (int32 feature_index = 0; feature_index < values_list.size();
111 ++feature_index) {
112 const Tensor& values_tensor = values_list[feature_index];
118 context, output_list.allocate(feature_index,
124 GetBuckets(feature_index, buckets_list);
139 GetFeatureDimension(feature_index, instance, indices_list);
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
H A Dhlo_parser.cc849 optional<int64> feature_index; local
850 attrs["feature_index"] = {/*required=*/true, AttrTy::kInt64,
851 &feature_index};
859 /*offset=*/operands[2], *epsilon, *feature_index));
865 optional<int64> feature_index; local
866 attrs["feature_index"] = {/*required=*/true, AttrTy::kInt64,
867 &feature_index};
876 /*variance=*/operands[4], *epsilon, *feature_index));
882 optional<int64> feature_index; local
883 attrs["feature_index"]
[all...]
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dcomputation_builder.cc1354 const ComputationDataHandle& offset, float epsilon, int64 feature_index) {
1362 request->set_feature_index(feature_index);
1369 const ComputationDataHandle& variance, float epsilon, int64 feature_index) {
1379 request->set_feature_index(feature_index);
1387 int64 feature_index) {
1396 request->set_feature_index(feature_index);
1352 BatchNormTraining( const ComputationDataHandle& operand, const ComputationDataHandle& scale, const ComputationDataHandle& offset, float epsilon, int64 feature_index) argument
1366 BatchNormInference( const ComputationDataHandle& operand, const ComputationDataHandle& scale, const ComputationDataHandle& offset, const ComputationDataHandle& mean, const ComputationDataHandle& variance, float epsilon, int64 feature_index) argument
1383 BatchNormGrad( const ComputationDataHandle& operand, const ComputationDataHandle& scale, const ComputationDataHandle& mean, const ComputationDataHandle& var, const ComputationDataHandle& grad_output, float epsilon, int64 feature_index) argument

Completed in 745 milliseconds