Searched refs:ValidateInputTensors (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 164 ValidateInputTensors(ctx, in0, in1);
215 virtual void ValidateInputTensors(OpKernelContext* ctx, const Tensor& in0,
231 void ValidateInputTensors(OpKernelContext* ctx, const Tensor& in0,
|
| D | banded_triangular_solve_op.cc | 219 ValidateInputTensors(ctx, in0, in1); in Compute()
269 void ValidateInputTensors(OpKernelContext* ctx, const Tensor& in0, in ValidateInputTensors() function in tensorflow::BandedTriangularSolveOpCpu
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 669 const Status s = ValidateInputTensors(ctx, in0, in1);
757 virtual Status ValidateInputTensors(OpKernelContext* ctx, const Tensor& in0,
778 Status ValidateInputTensors(OpKernelContext* ctx, const Tensor& in0,
822 Status ValidateInputTensors(OpKernelContext* ctx, const Tensor& in0,
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | rebatch_dataset_op.cc | 410 TF_RETURN_IF_ERROR(ValidateInputTensors()); in GetNextInternal()
570 Status ValidateInputTensors() TF_EXCLUSIVE_LOCKS_REQUIRED(mu_) { in ValidateInputTensors() function in tensorflow::data::experimental::__anon714a254d0111::RebatchDatasetV2Op::Dataset::Iterator
|