Home
last modified time | relevance | path

Searched defs:rank (Results 1 – 25 of 399) sorted by relevance

12345678910>>...16

/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dfft_ops_test.py42 def _compare(self, x, rank, fft_length=None, use_placeholder=False, argument
47 def _compare_forward(self, x, rank, fft_length=None, use_placeholder=False, argument
58 def _compare_backward(self, x, rank, fft_length=None, use_placeholder=False, argument
69 def _check_memory_fail(self, x, rank): argument
112 def _tf_fft(self, x, rank, fft_length=None, feed_dict=None): argument
117 def _tf_ifft(self, x, rank, fft_length=None, feed_dict=None): argument
122 def _np_fft(self, x, rank, fft_length=None): argument
132 def _np_ifft(self, x, rank, fft_length=None): argument
142 def _tf_fft_for_rank(self, rank): argument
152 def _tf_ifft_for_rank(self, rank): argument
[all …]
/external/clang/test/Analysis/
Dmpichecker.cpp6 int rank = 0; in matchedWait1() local
20 int rank = 0; in matchedWait2() local
33 int rank = 0; in matchedWait3() local
58 int rank = 0; in missingWait2() local
72 int rank = 0; in doubleNonblocking() local
86 int rank = 0; in doubleNonblocking2() local
100 int rank = 0; in doubleNonblocking3() local
110 int rank = 0; in doubleNonblocking4() local
131 int rank = 0; in missingNonBlocking() local
138 int rank = 0; in missingNonBlocking2() local
[all …]
DMemRegion.cpp8 int rank = 0; in testGetDescriptiveName() local
15 int rank = 0; in testGetDescriptiveName2() local
22 int rank = 0; in testGetDescriptiveName3() local
31 int rank = 0; in testGetDescriptiveName4() local
40 int rank = 0; in testGetDescriptiveName5() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_tensor_helper.h37 inline bool IsOfRankedFloatTensorType(RankedTensorType type, int rank) { in IsOfRankedFloatTensorType()
44 inline bool IsOfRankOrUnranked(Value value, int64_t rank) { in IsOfRankOrUnranked()
51 inline bool HasRankAtLeast(Value value, int64_t rank) { in HasRankAtLeast()
58 inline bool HasRankAtMost(Value value, int64_t rank) { in HasRankAtMost()
69 inline int64_t GetDimForAxis(int64_t axis, int64_t rank) { in GetDimForAxis()
/external/marisa-trie/lib/marisa/grimoire/vector/
Dbit-vector.cc352 const RankIndex &rank = ranks_[i / 512]; in rank1() local
416 const RankIndex &rank = ranks_[rank_id]; in select0() local
478 const RankIndex &rank = ranks_[rank_id]; in select1() local
518 const RankIndex &rank = ranks_[i / 512]; in rank1() local
585 const RankIndex &rank = ranks_[rank_id]; in select0() local
676 const RankIndex &rank = ranks_[rank_id]; in select1() local
/external/tensorflow/tensorflow/core/ops/
Dcount_ops.cc35 auto rank = c->Rank(output); in DenseCountSparseOutputShapeFn() local
46 auto rank = c->Dim(c->input(0), 1); in SparseCountSparseOutputShapeFn() local
55 int32_t rank = c->Rank(c->input(1)); in RaggedCountSparseOutputShapeFn() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h51 int32_t rank) { in UnchangedShapeWithRank()
60 shape_inference::InferenceContext* c, int32_t rank) { in UnchangedShapeWithRankAtLeast()
69 int32_t rank) { in UnchangedShapeWithRankAtMost()
Dshape_inference.cc262 const auto rank = Rank(s); in NumElements() local
315 Status InferenceContext::WithRank(ShapeHandle shape, int64_t rank, in WithRank()
340 Status InferenceContext::WithRankAtLeast(ShapeHandle shape, int64_t rank, in WithRankAtLeast()
355 Status InferenceContext::WithRankAtMost(ShapeHandle shape, int64_t rank, in WithRankAtMost()
444 const int32_t rank = Rank(prefix); in MergePrefix() local
472 const int32_t rank = Rank(s_old); in Relax() local
521 const int32_t rank = Rank(s0); in Merge() local
590 const int32_t rank = Rank(s); in Subshape() local
652 const int32_t rank = s1_rank + s2_rank; in Concatenate() local
700 ShapeHandle InferenceContext::UnknownShapeOfRank(int64_t rank) { in UnknownShapeOfRank()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc117 for (int rank = 0; rank < num_ranks; ++rank) { in Init() local
144 for (int rank = 0; rank < num_ranks; ++rank) { in Init() local
149 for (int rank = 0; rank < num_ranks; ++rank) { in Init() local
203 for (int rank = 0; rank < num_ranks; ++rank) { in RunTest() local
210 for (int rank = 0; rank < instances_.size(); ++rank) { in RunTest() local
251 DeviceInstance(int rank, const string& device_name, NcclTestBase* parent) in DeviceInstance()
456 void InitInput(Tensor* input, const int rank) override { in InitInput()
468 for (int rank = 0; rank < num_ranks; ++rank) { in InitExpected() local
491 void InitInput(Tensor* input, const int rank) override { in InitInput()
525 void InitInput(Tensor* input, const int rank) override { in InitInput()
[all …]
Dsparse_split_op_gpu.cu.cc109 Index input_nnz, int rank, int axis, SliceIndexer<Index> slice_indexer, in SparseSplitSliceIndexesKernel()
120 int rank, int axis, in LaunchSparseSplitSliceIndexesKernel()
162 Index input_nnz, int rank, int axis, SliceIndexer<Index> slice_indexer, in SparseSplitScatterKernel()
192 const GPUDevice& device, Index input_nnz, int rank, int axis, in LaunchSparseSplitScatterKernel()
220 const int rank = dense_shape.dims(); in operator ()() local
Ddeserialize_sparse_string_op.cc114 int rank = output_indices.dim_size(1); in Compute() local
163 int rank = shape.dims(); in Compute() local
242 int rank = output_indices->dim_size(1); in GetAndValidateSparseTensor() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc118 int rank = tensor_shape.dims(); in TensorShapeToXLAShape() local
154 int rank = tensor_shape.dims(); in TensorShapeToXLAShape() local
189 int64_t rank = shape.rank(); in GetShapeWithLayout() local
211 int64_t rank = input_shape.rank(); in GetShapeWithLayout() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/
Dcodegen_utils.cc31 int rank = memref.getType().cast<MemRefType>().getRank(); in emitNumElementsComputation() local
52 int rank = shape.size(); in calcMultiDimIndex() local
86 int rank = memref.getType().cast<MemRefType>().getRank(); in calcMultiDimIndex() local
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor.py250 def from_fields_and_rank(cls, fields, rank, validate=False): argument
548 def rank(self): member in StructuredTensor
932 def _from_pylist_of_dict(cls, pyval, keys, rank, typespec, path_so_far): argument
965 def _from_pylist_of_empty_dict(cls, pyval, rank): argument
1123 def _from_fields_and_rank(cls, fields, rank): argument
1167 def rank(self): member in StructuredTensor.Spec
1282 def _merge_row_partitions(row_partitions, value, rank, dtype, validate): argument
1304 def _row_partitions_for_tensor(value, rank, dtype): argument
1310 def _row_partitions_for_ragged_tensor(value, rank, dtype): argument
1321 def _row_partitions_for_uniform_shape(shape, rank): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_slice_ops.cc31 absl::InlinedVector<xla::XlaOp, 4> SliceVector(xla::XlaOp input, int64_t rank) { in SliceVector()
53 int64_t rank = input_shape.dims(); in Compile() local
Dshape_op.cc47 const int rank = input_shape.dims(); in Compile() local
184 const int rank = input_shape.dims(); in Compile() local
219 const int rank = input_shape.dims(); in Compile() local
241 const int rank = input_shape.dims(); in Compile() local
314 int64_t rank = shape.rank(); in Compile() local
/external/guava/android/guava/src/com/google/common/collect/
DExplicitOrdering.java43 private int rank(T value) { in rank() method in ExplicitOrdering
44 Integer rank = rankMap.get(value); in rank() local
/external/guava/guava/src/com/google/common/collect/
DExplicitOrdering.java43 private int rank(T value) { in rank() method in ExplicitOrdering
44 Integer rank = rankMap.get(value); in rank() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dtranspose_op.cc55 const int rank = input.dense_shape_host.size(); in ValidateTransposeInputs() local
150 const int rank = input_matrix.dims(); in operator ()() local
220 const int rank = input.dense_shape_host.size(); in operator ()() local
263 const int rank = x.dense_shape_host.size(); in operator ()() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_scatter_utils.cc29 int64_t rank = indices->shape().rank(); in TransformStartIndices() local
87 size_t dimension, size_t rank) { in MoveDimensionToEnd()
/external/turbine/java/com/google/turbine/tree/
DTurbineOperatorKind.java90 private final int rank; field in TurbineOperatorKind.Precedence
92 public int rank() { in rank() method in TurbineOperatorKind.Precedence
96 Precedence(int rank) { in Precedence()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc177 ScopedDataFormatUpgrader(TransposeContext* context, int rank) in ScopedDataFormatUpgrader()
731 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
770 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
794 const int rank = GetFaninPortRank(*node, 0); in TransposeNode() local
872 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
892 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
915 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
967 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
1153 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
1173 const int rank = GetFanoutPortRank(*node, 0); in TransposeNode() local
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dsharding_util_ops_test.py294 def testRanked(self, graph_fn, rank): argument
565 def testRanked(self, graph_fn, rank): argument
643 def testNoPadding(self, graph_fn, rank): argument
671 def testPartialPadding(self, graph_fn, rank): argument
700 def testCompletePadding(self, graph_fn, rank): argument
/external/eigen/test/
Dqr_colpivoting.cpp23 Index rank = internal::random<Index>(1, (std::min)(rows, cols) - 1); in cod() local
74 int rank = internal::random<int>(1, (std::min)(int(Rows), int(Cols)) - 1); in cod_fixedsize() local
106 Index rank = internal::random<Index>(1, (std::min)(rows, cols)-1); in qr() local
168 int rank = internal::random<int>(1, (std::min)(int(Rows), int(Cols))-1); in qr_fixedsize() local
/external/cldr/tools/cldr-code/src/main/java/org/unicode/cldr/test/
DEmojiSubdivisionNames.java132 int rank = LanguageGroup.rankInGroup(ulocale); in main() local
152 int rank = LanguageGroup.rankInGroup(ulocale); in main() local

12345678910>>...16