Searched refs:ValidateInputs (Results 1 – 4 of 4) 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()
157 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 | 49 ValidateInputs(rt_nested_splits, rt_dense_values_in)); in Compute()
152 static ::tensorflow::Status ValidateInputs( in ValidateInputs() function in tensorflow::RaggedTensorToSparseOp
|