Home
last modified time | relevance | path

Searched defs:cond_b (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc310 SpmdBuilder cond_b("fft_collective_permute_condition", hlo); in GetFinalFftUsingCollectivePermute() local
H A Ddot_handler.cc1753 SpmdBuilder cond_b("windowed_dot_general_cond", original_hlo); in PartitionBaseCase() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass_test.cc623 Output cond_b = ops::Placeholder(root.WithOpName("cond_b"), DT_BOOL); in TEST() local
661 Output cond_b = ops::Placeholder(root.WithOpName("cond_b"), DT_BOOL); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc213 Output cond_b = ops::Placeholder(scope.WithOpName("cond_b"), DT_BOOL); in TEST_F() local