Searched defs:expected_rank (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_rank_op_test.py | 77 def testRaggedRank(self, test_input, expected_rank, ragged_rank=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_variant_op.cc | 304 int index, int expected_rank, const Tensor** output_indices, in GetAndValidateSparseTensorIndicesAndValues()
|
| D | set_kernels.cc | 110 const auto expected_rank = sparse_tensor_shape.size(); in CheckGroup() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sharding_util_ops.cc | 72 const int expected_rank = num_partitions.size(); in GetAndValidateAttributes() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops.cc | 81 const int expected_rank = num_partitions.size(); in GetAndValidateAttributesHelper() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesLinearAlgebra.cpp | 260 int64_t expected_rank, in expect_at_least_rank()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device.cc | 788 int expected_rank, TF_DataType* expected_dtype, in VerifyTensorRankAndDType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 271 auto get_const_int = [](Value value, int expected_rank) -> Optional<int64_t> { in fold()
|