| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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;
[all...] |
| hlo_verifier.cc | 347 batch_norm_training->feature_index()));
359 batch_norm_inference->feature_index()));
369 batch_norm_grad->feature_index()));
[all...] |
| user_computation.cc | 533 offset->output_shape(), batch_norm_training_request.feature_index()));
579 batch_norm_inference_request.feature_index()));
625 batch_norm_grad_request.feature_index()));
[all...] |
| /external/harfbuzz_ng/src/ |
| hb-ot-layout-common-private.hh | [all...] |
| hb-ot-layout-private.hh | 46 unsigned int *feature_index);
|
| hb-ot-layout-gsubgpos-private.hh | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.h | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser_test.cc | 547 ROOT %batch-norm-training = (f32[2,2,1,2]{3,2,1,0}, f32[2]{0}, f32[2]{0}) batch-norm-training(f32[2,2,1,2]{3,2,1,0} %constant, f32[2]{0} %constant.1, f32[2]{0} %constant.2), epsilon=0.001, feature_index=3
563 ROOT %batch-norm-inference = f32[2,2,2,2]{3,2,1,0} batch-norm-inference(f32[2,2,2,2]{3,2,1,0} %input, f32[2]{0} %offset, f32[2]{0} %scale, f32[2]{0} %mean, f32[2]{0} %variance), epsilon=0.001, feature_index=0
579 ROOT %batch-norm-grad = (f32[2,2,2,2]{3,2,1,0}, f32[2]{0}, f32[2]{0}) batch-norm-grad(f32[2,2,2,2]{3,2,1,0} %input, f32[2]{0} %scale, f32[2]{0} %mean, f32[2]{0} %variance, f32[2,2,2,2]{3,2,1,0} %grad_output), epsilon=0.001, feature_index=0
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| example_proto_fast_parsing.cc | [all...] |
