Searched refs:compute_tuple (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 461 TF_ASSERT_OK_AND_ASSIGN(auto compute_tuple, builder.Build()); in TEST_F()
467 ReduceWindow({input1, input2}, {init, init}, compute_tuple, {4, 5}, {4, 5}, in TEST_F()
|
| D | hlo_evaluator_test.cc | 2888 auto compute_tuple = m_->AddEmbeddedComputation(bcompute.Build()); in TEST_P() local
2906 compute_tuple->ComputeProgramShape())); in TEST_P()
2908 shape, input_vec, init_vec, window, compute_tuple)); in TEST_P()
2939 auto compute_tuple = m_->AddEmbeddedComputation(bcompute.Build()); in TEST_P() local
2957 compute_tuple->ComputeProgramShape())); in TEST_P()
2959 shape, input_vec, init_vec, window, compute_tuple)); in TEST_P()
|