/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
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/tensorflow/tensorflow/compiler/xla/service/ |
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...] |
shape_inference.h | 94 int64 feature_index); 101 const Shape& variance_shape, int64 feature_index); 109 int64 feature_index);
|
hlo.proto | 109 int64 feature_index = 25;
|
shape_inference.cc | [all...] |
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); 832 int64 feature_index() const { return feature_index_; } function in class:xla::HloInstruction [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
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);
|
cudnn_batchnorm_thunk.cc | 36 MakeDescriptors(const Shape& shape, int64 feature_index) { 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),
|
cudnn_batchnorm_thunk.h | 53 float epsilon, int64 feature_index, 81 const BufferAllocation::Slice& offset, float epsilon, int64 feature_index, 114 float epsilon, int64 feature_index,
|
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},
|
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 = custom_call->operand(6) local [all...] |
/external/tensorflow/tensorflow/compiler/xla/tests/ |
batch_normalization_test.cc | 364 int64 feature_index; member in struct:xla::__anon39038::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/harfbuzz_ng/src/ |
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 = index 693 unsigned int feature_index; local [all...] |
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];
|
hb-ot-layout.h | 148 unsigned int *feature_index); 155 unsigned int *feature_index, 182 unsigned int *feature_index); 187 unsigned int feature_index, 252 unsigned int feature_index,
|
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);
|
hb-ot-map-private.hh | 203 unsigned int feature_index,
|
/external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
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/contrib/tensor_forest/kernels/ |
tree_utils.h | 41 const int32 FEATURE_INDEX = 1; 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/ |
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/contrib/boosted_trees/kernels/ |
quantile_ops.cc | 87 int32 GetFeatureDimension(const int32 feature_index, const int64 instance, 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); [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
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/ |
xla_data.proto | 618 int64 feature_index = 5; 628 int64 feature_index = 7; 638 int64 feature_index = 7;
|
/external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
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"] = {/*required=*/true, AttrTy::kInt64 [all...] |
/external/tensorflow/tensorflow/compiler/xla/client/ |
computation_builder.cc | [all...] |