| /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);
|
| 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/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/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/harfbuzz_ng/src/ |
| 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.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...] |
| /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/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/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...] |
| 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/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/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...] |
