/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | cudnn_batchnorm_rewriter.cc | 71 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 D | gpu_layout_assignment_test.cc | 135 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 D | cudnn_batchnorm_thunk.cc | 36 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 D | ir_emitter_unnested.cc | 323 // 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 D | decode_libsvm_op.cc | 70 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 D | batch_norm_op.cc | 56 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 D | hb-ot-map.cc | 85 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 D | hb-ot-shape-complex-arabic-fallback.hh | 50 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 D | hb-ot-layout.cc | 416 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 D | hb-ot-layout-gsubgpos-private.hh | 2326 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 D | tree_utils.h | 229 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 D | tensor_format.h | 420 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 D | sdca_internal.cc | 164 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 D | batchnorm_expander.cc | 165 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 D | hlo_instruction.h | 383 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 D | shape_inference.cc | 1181 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 D | hlo_instruction.cc | 143 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 D | batch_normalization_test.cc | 364 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 D | quantile_ops.cc | 87 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 D | hlo_parser.cc | 849 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 D | computation_builder.cc | 1354 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
|