Searched refs:CreateScalarOrComputation (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 106 XlaComputation CreateScalarOrComputation(PrimitiveType type, in CreateScalarOrComputation() function
117 XlaComputation logical_or = CreateScalarOrComputation(PRED, builder); in Any()
|
| D | arithmetic.h | 52 XlaComputation CreateScalarOrComputation(PrimitiveType type,
|
| D | matrix.cc | 71 ? CreateScalarOrComputation(shape.element_type(), builder) in GetMatrixDiagonal()
|
| D | self_adjoint_eig.cc | 235 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
|
| D | svd.cc | 614 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 133 reduce = CreateScalarOrComputation(PRED, &builder); in RunR1ToR0PredTest()
171 reduce_op = CreateScalarOrComputation(PRED, &builder); in RunR2ToR1PredTest()
803 return CreateScalarOrComputation(PRED, builder); in XLA_TEST_F()
976 auto reducer = CreateScalarOrComputation(U64, &builder); in XLA_TEST_F()
|