Home
last modified time | relevance | path

Searched defs:num_branches (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_case_op.cc278 int num_branches = branches_attr->list().func_size(); in RewriteCaseNode() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcase_op.cc90 int num_branches = branches.size(); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconditional_test.cc193 int num_branches = GetParam(); in XLA_TEST_P() local
236 int num_branches = GetParam(); in XLA_TEST_P() local
411 int num_branches = GetParam(); in XLA_TEST_P() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc546 int num_branches = instruction->branch_count(); in ImportInstructionImpl() local
/external/vixl/test/
Dtest-pool-manager.cc655 int num_branches, in AddNBranches()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1287 const int num_branches = conditional->branch_count(); in HandleConditional() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2672 int num_branches = conditional->branch_count(); in HandleConditional() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc2994 auto num_branches = op.branches().size(); in Verify() local