Searched refs:rhs_param (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 75 auto rhs_param = Parameter(b.get(), parameter_count * 2 + 1, scalar_shape, in CreateScalarComparisonComputation() local
78 rhs_params.emplace_back(rhs_param); in CreateScalarComparisonComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 280 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F() local
284 Add(lhs_param, rhs_param); in XLA_TEST_F()
319 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F() local
323 Sub(lhs_param, rhs_param); in XLA_TEST_F()
342 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F() local
344 Lt(lhs_param, rhs_param); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis_test.cc | 2615 HloInstruction* rhs_param = computation->parameter_instruction(1); in TEST_F() local
2621 rhs_param, {}, triangular_solve, {})); in TEST_F()
|