Searched refs:ConcatScalars (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.h | 89 xla::XlaOp ConcatScalars(xla::XlaBuilder* builder,
|
| D | prng.cc | 29 xla::XlaOp ConcatScalars(xla::XlaBuilder* builder, in ConcatScalars() function
370 return ConcatScalars(u128[0].builder(), {u128[0], u128[1]}); in Uint128ToOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateful_random_ops.cc | 55 state = xla::ConcatScalars(key.builder(), {key, state}); in BitGen()
164 return ConcatScalars(builder, {state, key}); in StateAndKeyToVariable()
|
| D | stateless_random_ops.cc | 56 state = xla::ConcatScalars(key.builder(), {key, state}); in GetBitGeneratorForDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamism_inference_test.cc | 188 auto concat = ConcatScalars(&b, {c, p}); in TEST_F()
|