Searched refs:b_float (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_matmul_op_test.cc | 311 Tensor b_float(DT_FLOAT, {b_rows, b_cols}); in TEST_F() local
313 &b_float, in TEST_F()
325 Tensor b_quantized = FloatTensorToQuantized<quint8>(b_float, b_min, b_max); in TEST_F()
|
| D | matmul_op.cc | 499 Tensor a_float, b_float, out_float; in Compute() local
501 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DT_FLOAT, b.shape(), &b_float)); in Compute()
508 BFloat16ToFloat(b.flat<bfloat16>().data(), b_float.flat<float>().data(), in Compute()
512 ctx, a_float, b_float, dim_pair, &algorithms_, use_autotune_, in Compute()
|
| D | sparse_matmul_op.cc | 998 std::unique_ptr<Tensor> b_float; in Compute() local
1010 b_float.reset(new Tensor(DT_FLOAT, b.shape())); in Compute()
1012 b_float->flat<float>().data(), b.NumElements()); in Compute()
1013 right = b_float.get(); in Compute()
|