Searched refs:ValidateInputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 41 void ValidateInputs(bool is_save_op, OpKernelContext* context, in ValidateInputs() function
99 ValidateInputs(true /* is save op */, context, prefix, tensor_names, in Compute()
178 ValidateInputs(false /* not save op */, context, prefix, tensor_names, in Compute()
|
| D | sparse_tensor_dense_add_op.cc | 35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs() function
81 ctx, ValidateInputs<Index>(a_indices_t, a_values_t, a_shape_t, b)); in Compute()
|
| D | sparse_reduce_op.cc | 106 Status ValidateInputs(const Tensor *shape_t, const Tensor *reduction_axes_t) { in ValidateInputs() function
167 OP_REQUIRES_OK(ctx, ValidateInputs(shape_t, reduction_axes_t)); in Compute()
263 OP_REQUIRES_OK(ctx, ValidateInputs(shape_t, reduction_axes_t)); in Compute()
|
| D | ragged_tensor_to_sparse_kernel.cc | 51 ValidateInputs(rt_nested_splits, rt_dense_values_in)); in Compute()
154 static ::tensorflow::Status ValidateInputs( in ValidateInputs() function in tensorflow::RaggedTensorToSparseOp
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_cholesky_op.cc | 85 ValidateInputs(ctx, *input_matrix, input_permutation_indices, &batch_size, in Compute()
229 void ValidateInputs(OpKernelContext* ctx, in ValidateInputs() function in tensorflow::CSRSparseCholeskyCPUOp
|
| D | mat_mul_op.cc | 104 Status ValidateInputs(const CSRSparseMatrix& sparse_matrix_a, in ValidateInputs() function in tensorflow::CSRMatMulOp
176 OP_REQUIRES_OK(ctx, this->ValidateInputs(*sparse_matrix_a, matrix_b, &rank, in Compute()
507 this->ValidateInputs(*a_matrix, b_t, &rank, &batch_size)); in Compute()
|