Searched refs:in0_reshaped (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | banded_triangular_solve_op.cc | 232 Tensor in0_reshaped; in Compute() local
235 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})), in Compute()
264 ctx, in0_reshaped, in1_reshaped, adjoint_, lower_, bcast, in Compute()
|
| D | matrix_triangular_solve_op_impl.h | 177 Tensor in0_reshaped;
180 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})),
210 ctx, in0_reshaped, in1_reshaped, adjoint_, lower_, bcast,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 686 Tensor in0_reshaped;
689 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})),
729 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DT_FLOAT, in0_reshaped.shape(),
737 BFloat16ToFloat(in0_reshaped.flat<bfloat16>().data(),
739 in0_reshaped.NumElements());
750 LaunchBatchMatMul<Device, Scalar>::Launch(ctx, in0_reshaped, in1_reshaped,
|