Searched refs:CreateScalarOrComputation (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 85 XlaComputation CreateScalarOrComputation(PrimitiveType type, in CreateScalarOrComputation() function
95 ? CreateScalarOrComputation(type, builder) in CreateScalarIdentityWithZeroComputation()
104 XlaComputation logical_or = CreateScalarOrComputation(PRED, builder); in Any()
|
| D | arithmetic.h | 59 XlaComputation CreateScalarOrComputation(PrimitiveType type,
|
| D | self_adjoint_eig.cc | 235 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
|
| D | svd.cc | 604 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
|
| D | math.cc | 1706 CreateScalarOrComputation(PRED, body_builder)); in LentzThompsonBarnettAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cast_op.cc | 194 CreateScalarOrComputation(dst_type_, ctx->builder()), in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 134 reduce = CreateScalarOrComputation(PRED, &builder); in RunR1ToR0PredTest()
172 reduce_op = CreateScalarOrComputation(PRED, &builder); in RunR2ToR1PredTest()
804 return CreateScalarOrComputation(PRED, builder); in XLA_TEST_F()
977 auto reducer = CreateScalarOrComputation(U64, &builder); in XLA_TEST_F()
|