| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_batchnorm_rewriter.cc | 73 HloInstruction* feature_index = local
75 LiteralUtil::CreateR0(batch_norm->feature_index())));
80 operands.push_back(feature_index);
109 HloInstruction* feature_index = local
111 LiteralUtil::CreateR0(batch_norm->feature_index())));
116 operands.push_back(feature_index);
178 HloInstruction* feature_index = local
180 LiteralUtil::CreateR0(batch_norm->feature_index())));
198 operands.push_back(feature_index);
|
| gpu_layout_assignment_test.cc | 142 auto* feature_index = local
148 {operand, scale, offset, mean, variance, epsilon, feature_index},
212 auto* feature_index = local
217 batchnorm_shape, {operand, scale, offset, epsilon, feature_index},
290 auto* feature_index = local
298 feature_index},
|
| ir_emitter_unnested.cc | 379 // feature_index operands be constants.
386 const HloInstruction* feature_index = custom_call->operand(6); local
387 CHECK(feature_index->IsConstant());
388 int64 feature_index_value = feature_index->literal().Get<int64>({});
398 /*feature_index=*/feature_index_value,
410 const HloInstruction* feature_index = custom_call->operand(4); local
411 CHECK(feature_index->IsConstant());
412 int64 feature_index_value = feature_index->literal().Get<int64>({});
426 /*feature_index=*/feature_index_value,
440 const HloInstruction* feature_index = custom_call->operand(6) local
[all...] |
| /external/harfbuzz_ng/test/api/ |
| test-ot-name.c | 36 unsigned int feature_index; local
50 &feature_index))
53 if (!hb_ot_layout_feature_get_name_ids (face, HB_OT_TAG_GSUB, feature_index,
64 all_chars = hb_ot_layout_feature_get_characters (face, HB_OT_TAG_GSUB, 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/harfbuzz_ng/src/ |
| hb-ot-map.cc | 96 unsigned int feature_index,
114 feature_index,
223 unsigned int feature_index[2]; local
234 &feature_index[table_index]);
243 &feature_index[table_index]);
253 map->index[0] = feature_index[0];
254 map->index[1] = feature_index[1];
|
| hb-ot-layout.cc | 461 unsigned int *feature_index)
470 if (feature_index) *feature_index = i;
475 if (feature_index) *feature_index = HB_OT_LAYOUT_NO_FEATURE_INDEX;
544 unsigned int *feature_index)
550 feature_index,
564 unsigned int *feature_index,
571 if (feature_index) *feature_index = index
761 unsigned int feature_index = l.get_feature_index (i); local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batch_norm_op.cc | 53 int feature_index = variable
63 input, ctx->Input(1), ctx->Input(2), epsilon_, feature_index);
92 epsilon_, feature_index);
142 const int feature_index = variable
167 epsilon_, feature_index);
175 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index,
177 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(),
178 feature_index + 1);
198 xla::Mul(grad_backprop, xla::Sub(activations, mean, {feature_index}));
206 xla::Mul(grad_backprop, xla::Mul(scratch1, scale), {feature_index});
[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/kernels/ |
| sdca_internal.cc | 170 const int64 feature_index = (*sparse_features.indices)(k); local
175 const float sparse_weight = sparse_weights.nominals(l, feature_index);
178 sparse_weights.deltas(l, feature_index) * num_loss_partitions;
431 const int64 feature_index = (*sparse_features->indices)(k);
432 if (!weights.SparseIndexValid(i, feature_index)) {
518 const int64 feature_index = (*sparse_features.indices)(k);
519 if (previous_indices.insert(feature_index).second == false) {
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_format.h | 527 int feature_index = GetTensorFeatureDimIndex(dims, format); local
534 dim_sizes[feature_index] = C;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander.cc | 113 HloInstruction* operand, int64 feature_index,
120 if (i == feature_index) {
208 int64 feature_index = batch_norm->feature_index(); local
226 if (i != feature_index) {
232 add(DynamicElementCountPerFeature(operand, feature_index, add));
235 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index}));
238 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index}));
260 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index}));
274 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index}));
334 int64 feature_index = batch_norm->feature_index(); local
470 int64 feature_index = batch_norm->feature_index(); local
[all...] |
| hlo_instructions.h | 28 // Returns feature_index field associated with the instruction. The index
30 int64 feature_index() const { return feature_index_; } function in class:xla::HloBatchNormInstruction
43 int64 feature_index);
65 float epsilon, int64 feature_index);
79 float epsilon, int64 feature_index);
93 HloInstruction* grad_output, float epsilon, int64 feature_index);
[all...] |
| hlo_parser.cc | 1317 optional<int64> feature_index; local
1333 optional<int64> feature_index; local
1350 optional<int64> feature_index; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| batch_normalization_test.cc | 373 int64 feature_index; member in struct:xla::__anon44583::BatchNormTestParam
381 os << "feature_index=" << p.feature_index << ", ";
408 auto add_testcase = [&](std::vector<int64> bounds, int64 feature_index,
410 BatchNormTestParam p{bounds, feature_index, random_value_mean,
457 const int64 feature_index = GetParam().feature_index; local
459 Product(bounds) / bounds[feature_index];
460 const int64 feature_bound = bounds[feature_index];
468 if (i != feature_index) {
558 const int64 feature_index = GetParam().feature_index; local
670 const int64 feature_index = GetParam().feature_index; local
[all...] |
| /external/libaom/libaom/av1/encoder/ |
| encodeframe.c | 2643 int feature_index = 0; local
2795 int feature_index = 0; local
2882 int feature_index = 0; local
3058 int feature_index = 0; local
[all...] |
| /external/libvpx/libvpx/vp9/encoder/ |
| vp9_encodeframe.c | 3251 int feature_index = 0; local
3339 int feature_index = 0; local
[all...] |
