Home
last modified time | relevance | path

Searched refs:contraction (Results 1 – 25 of 64) sorted by relevance

123

/external/eigen/bench/tensors/
Dtensor_benchmarks_cpu.cc99 BM_FuncWithInputDimsCPU(contraction, N, N, N, 1);
100 BM_FuncWithInputDimsCPU(contraction, N, N, N, 4);
101 BM_FuncWithInputDimsCPU(contraction, N, N, N, 8);
102 BM_FuncWithInputDimsCPU(contraction, N, N, N, 12);
103 BM_FuncWithInputDimsCPU(contraction, N, N, N, 16);
105 BM_FuncWithInputDimsCPU(contraction, 64, N, N, 1);
106 BM_FuncWithInputDimsCPU(contraction, 64, N, N, 4);
107 BM_FuncWithInputDimsCPU(contraction, 64, N, N, 8);
108 BM_FuncWithInputDimsCPU(contraction, 64, N, N, 12);
109 BM_FuncWithInputDimsCPU(contraction, 64, N, N, 16);
[all …]
Dtensor_benchmarks_fp16_gpu.cu52 BM_FuncWithInputDimsGPU(contraction, N, N, N);
53 BM_FuncWithInputDimsGPU(contraction, 64, N, N);
54 BM_FuncWithInputDimsGPU(contraction, N, 64, N);
55 BM_FuncWithInputDimsGPU(contraction, N, N, 64);
Dtensor_benchmarks_gpu.cu52 BM_FuncWithInputDimsGPU(contraction, N, N, N);
53 BM_FuncWithInputDimsGPU(contraction, 64, N, N);
54 BM_FuncWithInputDimsGPU(contraction, N, 64, N);
55 BM_FuncWithInputDimsGPU(contraction, N, N, 64);
Dcontraction_benchmarks_cpu.cc19 suite.contraction(iters); \
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc114 ContractionWithBiasAdd(int contraction, int bias_add) in ContractionWithBiasAdd()
115 : contraction(contraction), bias_add(bias_add) {} in ContractionWithBiasAdd()
117 int contraction = kMissingIndex; member
124 ContractionWithBiasAddAndActivation(int contraction, int bias_add, in ContractionWithBiasAddAndActivation()
126 : contraction(contraction), bias_add(bias_add), activation(activation) {} in ContractionWithBiasAddAndActivation()
128 int contraction = kMissingIndex; member
136 ContractionWithSqueezeAndBiasAdd(int contraction, int squeeze, int bias_add) in ContractionWithSqueezeAndBiasAdd()
137 : contraction(contraction), squeeze(squeeze), bias_add(bias_add) {} in ContractionWithSqueezeAndBiasAdd()
139 int contraction = kMissingIndex; member
147 ContractionWithBatchNorm(int contraction, int fused_batch_norm, in ContractionWithBatchNorm()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfused_kernel_matcher.cc106 LogicalResult matchAndRewrite(SrcOpT contraction, in matchAndRewrite() argument
111 if (!contraction.getResult().hasOneUse()) in matchAndRewrite()
112 return rewriter.notifyMatchFailure(contraction, in matchAndRewrite()
115 BiasAddOp bias_add = GetBiasAdd(contraction.getResult()); in matchAndRewrite()
118 contraction, "does not feed into a tf.BiasAdd/tf.BiasAddV1 op"); in matchAndRewrite()
121 if (!AreFuseCompatible(contraction, bias_add, rewriter)) { in matchAndRewrite()
123 contraction, "cannot fuse with the subsequent BiasAdd op"); in matchAndRewrite()
126 SmallVector<Location, 3> locations{contraction.getLoc(), bias_add.getLoc()}; in matchAndRewrite()
152 SmallVector<Value, 4> operands(contraction.operand_begin(), in matchAndRewrite()
153 contraction.operand_end()); in matchAndRewrite()
[all …]
/external/llvm-project/polly/lib/External/ppcg/
Dgrouping.c44 isl_union_pw_multi_aff *contraction; member
54 isl_union_pw_multi_aff_free(grouping->contraction); in ppcg_grouping_clear()
500 grouping->contraction); in add_groups()
505 grouping->contraction = upma; in add_groups()
508 if (!grouping->domain || !grouping->contraction || in add_groups()
629 upma = isl_union_pw_multi_aff_union_add(upma, grouping->contraction); in complete_grouping()
630 grouping->contraction = upma; in complete_grouping()
651 isl_union_pw_multi_aff *contraction; in ppcg_compute_schedule() local
662 if (!grouping.contraction) { in ppcg_compute_schedule()
667 contraction = isl_union_pw_multi_aff_copy(grouping.contraction); in ppcg_compute_schedule()
[all …]
Dgpu.c1166 isl_union_pw_multi_aff_free(kernel->contraction); in ppcg_kernel_free()
3205 isl_union_pw_multi_aff_copy(kernel->contraction)); in anchored_non_local_accesses()
3342 isl_union_pw_multi_aff *contraction; in add_copies_group_private() local
3369 contraction = isl_union_pw_multi_aff_copy(kernel->contraction); in add_copies_group_private()
3370 filter = isl_union_set_preimage_union_pw_multi_aff(filter, contraction); in add_copies_group_private()
3713 isl_union_pw_multi_aff *contraction; in compute_sync_writes() local
3721 contraction = kernel->contraction; in compute_sync_writes()
3723 kernel_prefix, isl_union_pw_multi_aff_copy(contraction)); in compute_sync_writes()
3725 thread_prefix, isl_union_pw_multi_aff_copy(contraction)); in compute_sync_writes()
3842 isl_union_pw_multi_aff *contraction; in gpu_create_kernel() local
[all …]
Dgpu_group.c1503 isl_union_pw_multi_aff *contraction; in check_can_be_private_live_ranges() local
1511 contraction = isl_union_pw_multi_aff_copy(kernel->contraction); in check_can_be_private_live_ranges()
1513 contraction); in check_can_be_private_live_ranges()
1546 __isl_keep isl_union_pw_multi_aff *contraction) in expand() argument
1548 contraction = isl_union_pw_multi_aff_copy(contraction); in expand()
1549 s = isl_union_map_preimage_domain_union_pw_multi_aff(s, contraction); in expand()
1632 isl_union_pw_multi_aff *contraction; in gpu_group_references() local
1660 contraction = isl_union_pw_multi_aff_copy(kernel->contraction); in gpu_group_references()
1661 data.host_sched = expand(data.host_sched, contraction); in gpu_group_references()
1662 data.shared_sched = expand(data.shared_sched, contraction); in gpu_group_references()
[all …]
/external/llvm-project/polly/lib/External/isl/
Disl_schedule_node.c2037 isl_union_pw_multi_aff *contraction; in isl_schedule_node_band_sink() local
2061 contraction = isl_schedule_node_get_subtree_contraction(node); in isl_schedule_node_band_sink()
2066 tree = isl_schedule_tree_pullback_union_pw_multi_aff(tree, contraction); in isl_schedule_node_band_sink()
2142 __isl_take isl_union_pw_multi_aff *contraction, in isl_schedule_node_expansion_set_contraction_and_expansion() argument
2147 if (!node || !contraction || !expansion) in isl_schedule_node_expansion_set_contraction_and_expansion()
2152 contraction, expansion); in isl_schedule_node_expansion_set_contraction_and_expansion()
2156 isl_union_pw_multi_aff_free(contraction); in isl_schedule_node_expansion_set_contraction_and_expansion()
2543 __isl_take isl_union_pw_multi_aff *contraction, in isl_schedule_node_insert_expansion() argument
2552 tree = isl_schedule_tree_insert_expansion(tree, contraction, expansion); in isl_schedule_node_insert_expansion()
2819 isl_union_pw_multi_aff *contraction; member
[all …]
Disl_schedule_read.c171 isl_union_pw_multi_aff *contraction = NULL; in read_expansion() local
189 isl_union_pw_multi_aff_free(contraction); in read_expansion()
192 contraction = isl_union_pw_multi_aff_read_from_str(ctx, in read_expansion()
196 if (!contraction) in read_expansion()
224 if (!contraction) in read_expansion()
232 return isl_schedule_tree_from_expansion(contraction, expansion); in read_expansion()
233 return isl_schedule_tree_insert_expansion(tree, contraction, expansion); in read_expansion()
236 isl_union_pw_multi_aff_free(contraction); in read_expansion()
Disl_schedule_tree.c105 dup->contraction = in isl_schedule_tree_dup()
106 isl_union_pw_multi_aff_copy(tree->contraction); in isl_schedule_tree_dup()
108 if (!dup->contraction || !dup->expansion) in isl_schedule_tree_dup()
195 isl_union_pw_multi_aff_free(tree->contraction); in isl_schedule_tree_free()
311 __isl_take isl_union_pw_multi_aff *contraction, in isl_schedule_tree_from_expansion() argument
317 if (!contraction || !expansion) in isl_schedule_tree_from_expansion()
325 tree->contraction = contraction; in isl_schedule_tree_from_expansion()
330 isl_union_pw_multi_aff_free(contraction); in isl_schedule_tree_from_expansion()
658 tree1->contraction, tree2->contraction); in isl_schedule_tree_plain_is_equal()
911 __isl_take isl_union_pw_multi_aff *contraction, in isl_schedule_tree_insert_expansion() argument
[all …]
Disl_schedule_node_private.h48 __isl_take isl_union_pw_multi_aff *contraction,
62 __isl_take isl_union_pw_multi_aff *contraction,
Disl_schedule_tree.h64 isl_union_pw_multi_aff *contraction; member
97 __isl_take isl_union_pw_multi_aff *contraction,
160 __isl_take isl_union_pw_multi_aff *contraction,
205 __isl_take isl_union_pw_multi_aff *contraction,
/external/apache-commons-math/src/main/java/org/apache/commons/math/util/
DResizableDoubleArray.java389 protected void checkContractExpand(float contraction, float expansion) { in checkContractExpand() argument
391 if (contraction < expansion) { in checkContractExpand()
394 contraction, expansion); in checkContractExpand()
397 if (contraction <= 1.0) { in checkContractExpand()
400 contraction); in checkContractExpand()
/external/llvm-project/mlir/test/Dialect/Linalg/
Dcodegen-strategy.mlir1 // RUN: mlir-opt %s -test-linalg-codegen-strategy="tile-sizes=2,4,8 vectorize vectorize-contraction…
2 …te promote-full-tile-pad register-tile-sizes=2,4,8 vectorize vectorize-contraction-to=outerproduct…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_interfaces.td113 A contraction fusable operation is one that can be fused into the output of
114 a tensor contraction (MatMul, Conv2D, etc...) operation.
116 For example all element wise operations are trivially contraction fusable.
121 [{Returns contraction fusion if the operation satisfies all the fusion
/external/icu/android_icu4j/src/main/tests/android/icu/dev/data/
Dcollationtest.txt115 # Some strings are chosen to cause incremental contiguous contraction matching to
122 # contraction starter with no following text, or mismatch, or blocked
151 # no discontiguous contraction because of missing prefix contraction d+z,
233 # partial contiguous match, backs up, matches discontiguous contraction
250 # to match the discontiguous contraction of the first 0f71 with the trailing 0f72 inside the 0f73
712 ** test: contraction inside extension, ICU ticket 9378
857 ** test: tailor contraction together with nearly equivalent prefix, ICU ticket 10101
869 = opx # first contraction op, then prefix p|x
884 ** test: prefix+contraction together (1), ICU ticket 10071
904 ** test: prefix+contraction together (2), ICU ticket 10071
[all …]
/external/icu/icu4j/main/tests/collate/src/com/ibm/icu/dev/data/
Dcollationtest.txt115 # Some strings are chosen to cause incremental contiguous contraction matching to
122 # contraction starter with no following text, or mismatch, or blocked
151 # no discontiguous contraction because of missing prefix contraction d+z,
233 # partial contiguous match, backs up, matches discontiguous contraction
250 # to match the discontiguous contraction of the first 0f71 with the trailing 0f72 inside the 0f73
712 ** test: contraction inside extension, ICU ticket 9378
857 ** test: tailor contraction together with nearly equivalent prefix, ICU ticket 10101
869 = opx # first contraction op, then prefix p|x
884 ** test: prefix+contraction together (1), ICU ticket 10071
904 ** test: prefix+contraction together (2), ICU ticket 10071
[all …]
/external/icu/icu4c/source/test/testdata/
Dcollationtest.txt115 # Some strings are chosen to cause incremental contiguous contraction matching to
122 # contraction starter with no following text, or mismatch, or blocked
151 # no discontiguous contraction because of missing prefix contraction d+z,
233 # partial contiguous match, backs up, matches discontiguous contraction
250 # to match the discontiguous contraction of the first 0f71 with the trailing 0f72 inside the 0f73
712 ** test: contraction inside extension, ICU ticket 9378
857 ** test: tailor contraction together with nearly equivalent prefix, ICU ticket 10101
869 = opx # first contraction op, then prefix p|x
884 ** test: prefix+contraction together (1), ICU ticket 10071
904 ** test: prefix+contraction together (2), ICU ticket 10071
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Einsum.pbtxt21 summary: "Tensor contraction according to Einstein summation convention."
23 Implements generalized Tensor contraction and reduction. Each input Tensor must
48 contraction.
54 dimensions in Tensor contraction. Unnamed axis labels corresponding to
/external/icu/icu4c/source/test/intltest/
Ditercoll.cpp222 UnicodeString contraction = "change"; in TestOffset() local
229 iter = tailored.createCollationElementIterator(contraction); in TestOffset()
240 contraction = "peache"; in TestOffset()
241 iter = tailored.createCollationElementIterator(contraction); in TestOffset()
/external/icu/android_icu4j/src/main/tests/android/icu/dev/test/collator/
DCollationIteratorTest.java262 String contraction = "change"; in TestOffset() local
270 iter = tailored.getCollationElementIterator(contraction); in TestOffset()
277 contraction = "peache"; in TestOffset()
278 iter = tailored.getCollationElementIterator(contraction); in TestOffset()
/external/icu/icu4j/main/tests/collate/src/com/ibm/icu/dev/test/collator/
DCollationIteratorTest.java259 String contraction = "change"; in TestOffset() local
267 iter = tailored.getCollationElementIterator(contraction); in TestOffset()
274 contraction = "peache"; in TestOffset()
275 iter = tailored.getCollationElementIterator(contraction); in TestOffset()
/external/tensorflow/third_party/
Dopt_einsum.BUILD2 # opt_einsum is a library for optimizing tensor contraction order for einsum.

123