Searched refs:in1_reshaped (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | banded_triangular_solve_op.cc | 240 Tensor in1_reshaped; in Compute() local
243 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})), in Compute()
264 ctx, in0_reshaped, in1_reshaped, adjoint_, lower_, bcast, in Compute()
|
| D | matrix_triangular_solve_op_impl.h | 185 Tensor in1_reshaped;
188 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})),
210 ctx, in0_reshaped, in1_reshaped, adjoint_, lower_, bcast,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 694 Tensor in1_reshaped;
697 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})),
731 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DT_FLOAT, in1_reshaped.shape(),
740 BFloat16ToFloat(in1_reshaped.flat<bfloat16>().data(),
742 in1_reshaped.NumElements());
750 LaunchBatchMatMul<Device, Scalar>::Launch(ctx, in0_reshaped, in1_reshaped,
|