| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_utils_test.cc | 356 TF_EXPECT_OK((ValidateSparseTensor<int64_t>(indices, values, shape, in TEST_P()
372 EXPECT_THAT((ValidateSparseTensor<int64_t>(indices, values, shape, in TEST_P()
390 EXPECT_THAT((ValidateSparseTensor<int64_t>(indices, values, shape, in TEST_P()
408 EXPECT_THAT((ValidateSparseTensor<int64_t>(indices, values, shape, in TEST_P()
427 EXPECT_THAT((ValidateSparseTensor<int64_t>(indices, values, shape, in TEST_P()
440 (ValidateSparseTensor<int64_t>(indices, values, shape, in TEST_P()
469 Status indices_valid = ValidateSparseTensor<int64_t>( in TEST_P()
514 Status indices_valid = ValidateSparseTensor<int64_t>( in TEST_P()
563 ::benchmark::DoNotOptimize(ValidateSparseTensor<int64_t>( in BM_ValidateSparseTensor()
|
| H A D | sparse_utils.cc | 291 Status ValidateSparseTensor(const Tensor& indices, const Tensor& values, in ValidateSparseTensor() function
318 template Status ValidateSparseTensor<TypeIndex>( \
|
| H A D | sparse_utils.h | 81 Status ValidateSparseTensor(const Tensor& indices, const Tensor& values,
|
| H A D | bincount_op.cc | 395 OP_REQUIRES_OK(ctx, sparse_utils::ValidateSparseTensor<int64_t>( in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| H A D | multiplex_3_kernel.cc | 50 ValidateSparseTensor(cond_indices_tensor, cond_values_tensor, in Compute()
52 OP_REQUIRES_OK(ctx, ValidateSparseTensor(a_indices_tensor, a_values_tensor, in Compute()
54 OP_REQUIRES_OK(ctx, ValidateSparseTensor(b_indices_tensor, b_values_tensor, in Compute()
179 Status ValidateSparseTensor(const ::tensorflow::Tensor& indices_tensor, in ValidateSparseTensor() function in tensorflow::custom_op_examples::MultiplexSparseOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns_test.cc | 1369 TF_EXPECT_OK(ValidateSparseTensor(&c, indices, values, shape)); in TEST()
1382 TF_EXPECT_OK(ValidateSparseTensor(&c, indices, values, shape)); in TEST()
1396 ValidateSparseTensor(&c, indices, values, shape).code()); in TEST()
1410 ValidateSparseTensor(&c, indices, values, shape).code()); in TEST()
1424 ValidateSparseTensor(&c, indices, values, shape).code()); in TEST()
1437 TF_EXPECT_OK(ValidateSparseTensor(&c, indices, values, shape)); in TEST()
1450 TF_EXPECT_OK(ValidateSparseTensor(&c, indices, values, shape)); in TEST()
1463 TF_EXPECT_OK(ValidateSparseTensor(&c, indices, values, shape)); in TEST()
1476 TF_EXPECT_OK(ValidateSparseTensor(&c, indices, values, shape)); in TEST()
1479 TEST(CommonShapeFnsTest, ValidateSparseTensor) { in TEST() argument
[all …]
|
| H A D | common_shape_fns.h | 257 Status ValidateSparseTensor(InferenceContext* c, ShapeHandle indices_shape,
|
| H A D | common_shape_fns.cc | 2318 Status ValidateSparseTensor(InferenceContext* c, ShapeHandle indices_shape, in ValidateSparseTensor() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | set_ops.cc | 110 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anona97893f50202()
157 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anona97893f50302()
159 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anona97893f50302()
|
| H A D | sparse_csr_matrix_ops.cc | 66 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anona88575120102()
140 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anona88575120302()
|
| H A D | array_ops.cc | 1063 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anon38bbb0e81502()
1065 TF_RETURN_IF_ERROR(shape_inference::ValidateSparseTensor( in __anon38bbb0e81502()
|