Home
last modified time | relevance | path

Searched defs:oidx (Results 1 – 6 of 6) 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/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