Searched refs:key_op (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | rng_bit_generator_expander.cc | 67 XlaOp key_op = Reshape(Slice(state_param, {0}, {1}, {1}), {}); in GetGeneratorComputation() local 71 output = ThreeFryBitGenerator(key_op, Slice(state_param, {1}, {2}, {1}), in GetGeneratorComputation() 76 key_op, GetPhiloxStateOp(state_param, state_shape), data_shape); in GetGeneratorComputation() 85 ConcatInDim(&builder, {Reshape(key_op, {1}), output.state}, 0); in GetGeneratorComputation()
|
/external/rust/crates/quiche/deps/boringssl/src/crypto/evp/ |
D | evp_test.cc | 380 int (*key_op)(EVP_PKEY_CTX *ctx, uint8_t *out, size_t *out_len, in TestEVP() local 387 key_op = EVP_PKEY_decrypt; in TestEVP() 390 key_op = EVP_PKEY_sign; in TestEVP() 401 key_op = EVP_PKEY_encrypt; in TestEVP() 495 if (!key_op(pctx, nullptr, &len, input.data(), input.size())) { in TestEVP() 499 if (!key_op(pctx, actual.data(), &len, input.data(), input.size())) { in TestEVP()
|
/external/boringssl/src/crypto/evp/ |
D | evp_test.cc | 380 int (*key_op)(EVP_PKEY_CTX *ctx, uint8_t *out, size_t *out_len, in TestEVP() local 387 key_op = EVP_PKEY_decrypt; in TestEVP() 390 key_op = EVP_PKEY_sign; in TestEVP() 401 key_op = EVP_PKEY_encrypt; in TestEVP() 495 if (!key_op(pctx, nullptr, &len, input.data(), input.size())) { in TestEVP() 499 if (!key_op(pctx, actual.data(), &len, input.data(), input.size())) { in TestEVP()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | tpu_rewrite_pass.cc | 642 parallel_op.walk([&](TF::_TPUCompileMlirPlaceholderProgramKeyOp key_op) { in Rewrite() argument 643 key_op.replaceAllUsesWith(compile_op->getResult(1)); in Rewrite() 644 key_op.erase(); in Rewrite()
|