Home
last modified time | relevance | path

Searched defs:oidx (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dintroduce_floating_point_jitter_pass.cc62 Graph* g, Node* n, int oidx, float jitter_amount, in IntroduceJitterToTensor()
136 for (int oidx : p.second) { in IntroduceFloatingPointJitter() local
Dnode_matchers.cc527 int oidx, ::testing::Matcher<const Node*> node_matcher) { in Out()
Dbuild_xla_ops_pass.cc130 int oidx = e->src_output(); in MergeOutgoingDataEdges() local
/external/libopenapv/app/
Doapv_app_args.h82 int oidx = 0; in args_search_long_key() local
98 int oidx = 0; in args_search_short_arg() local
217 int oidx; in args_parse_cfg() local
260 int oidx; /* option index */ in args_parse_cmd() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_degenerate_dim_remover.cc102 for (int oidx = 0; oidx < instr->input_count(); oidx++) { in HandleReduce() local
Dreduction_layout_normalizer.cc166 for (int oidx = 0; oidx < reduce->input_count(); oidx++) { in HandleReduce() local
Dir_emitter_unnested.cc3877 for (int oidx = 0; oidx < num_outputs; oidx++) { in EmitReductionOutputForRowReduction() local
3903 for (int oidx = 0; oidx < num_outputs; oidx++) { in EmitReductionOutputForRowReduction() local
3919 for (int oidx = 0; oidx < num_outputs; oidx++) { in EmitReductionOutputForRowReduction() local
4037 for (int oidx = 0; oidx < num_outputs; oidx++) { in EmitReductionOutputForColumnReduction() local
/external/pytorch/aten/src/ATen/native/nested/cuda/
DNestedTensorTransformerFunctions.cu657 const int oidx = outer / jagged_folded_size; in __launch_bounds__() local
847 const int oidx = offset_temp; in __launch_bounds__() local
/external/pytorch/functorch/csrc/dim/
Ddim.cpp1210 auto oidx = from_levels.index(l); in _match_levels() local