Searched refs:indexed_case (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| D | case_op.pbtxt | 44 name: "indexed_case"
101 name: "indexed_case/Identity"
103 input: "indexed_case"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | case_op.pbtxt | 46 name: "indexed_case"
103 name: "indexed_case/Identity"
105 input: "indexed_case"
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/graphdef_to_mlir/ |
| D | import_case_with_attr_inference.pbtxt | 49 name: "indexed_case"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_code_motion_test.cc | 1321 …1,0}, s32[]) %conditional.3), index=0, metadata={op_type="Case" op_name="switch_case/indexed_case"} in TEST_F()
1325 …1,0}, s32[]) %conditional.3), index=1, metadata={op_type="Case" op_name="switch_case/indexed_case"} in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | cond_v2_test.py | 993 return cond_v2.indexed_case(
1720 output = cond_v2.indexed_case(
1736 output = cond_v2.indexed_case(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 1024 def indexed_case(branch_index, function
|
| D | control_flow_ops.py | 3343 return cond_v2.indexed_case(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 2859 return cond_v2.indexed_case(
2881 return cond_v2.indexed_case(case_input,
|