| /external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/ |
| D | MLInlineAdvisor.cpp | 347 *ModelRunner->getTensor<int64_t>(FeatureIndex::CalleeBasicBlockCount) = in getAdviceImpl()
349 *ModelRunner->getTensor<int64_t>(FeatureIndex::CallSiteHeight) = in getAdviceImpl()
351 *ModelRunner->getTensor<int64_t>(FeatureIndex::NodeCount) = NodeCount; in getAdviceImpl()
352 *ModelRunner->getTensor<int64_t>(FeatureIndex::NrCtantParams) = NrCtantParams; in getAdviceImpl()
353 *ModelRunner->getTensor<int64_t>(FeatureIndex::EdgeCount) = EdgeCount; in getAdviceImpl()
354 *ModelRunner->getTensor<int64_t>(FeatureIndex::CallerUsers) = in getAdviceImpl()
356 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
359 *ModelRunner->getTensor<int64_t>(FeatureIndex::CallerBasicBlockCount) = in getAdviceImpl()
361 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
364 *ModelRunner->getTensor<int64_t>(FeatureIndex::CalleeUsers) = in getAdviceImpl()
[all …]
|
| /external/tensorflow/tensorflow/tools/android/inference_interface/java/org/tensorflow/contrib/android/ |
| D | TensorFlowInferenceInterface.java | 488 getTensor(outputName).writeTo(dst); in fetch()
498 getTensor(outputName).writeTo(dst); in fetch()
508 getTensor(outputName).writeTo(dst); in fetch()
518 getTensor(outputName).writeTo(dst); in fetch()
528 getTensor(outputName).writeTo(dst); in fetch()
609 private Tensor<?> getTensor(String outputName) { in getTensor() method in TensorFlowInferenceInterface
|
| /external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/Analysis/ |
| D | MLModelRunner.h | 35 template <typename T, typename I> T *getTensor(I FeatureID) { in getTensor() function
40 template <typename T, typename I> const T *getTensor(I FeatureID) const { in getTensor() function
|
| /external/swiftshader/third_party/llvm-16.0/llvm/lib/CodeGen/ |
| D | MLRegallocPriorityAdvisor.cpp | 277 *Runner->getTensor<int64_t>(0) = static_cast<int64_t>(Size); in getPriorityImpl()
278 *Runner->getTensor<int64_t>(1) = static_cast<int64_t>(Stage); in getPriorityImpl()
279 *Runner->getTensor<float>(2) = static_cast<float>(LI.weight()); in getPriorityImpl()
|
| D | MLRegallocEvictAdvisor.cpp | 752 Runner->getTensor<float>(FeatureIndex)[Pos] /= Largest[FeatureIndex]; in tryFindEvictionCandidate()
755 *Runner->getTensor<float>(FeatureIDs::progress) = in tryFindEvictionCandidate()
897 Runner->getTensor<TYPE>(FeatureIDs::ID)[Pos] = static_cast<TYPE>(VAL); \ in extractFeatures()
990 RegallocRunner->getTensor<int64_t>(InstructionsIndex)[InstructionIndex] = in extractInstructionFeatures()
994 RegallocRunner->getTensor<int64_t>( in extractInstructionFeatures()
1012 RegallocRunner->getTensor<int64_t>( in extractInstructionFeatures()
1052 RegallocRunner->getTensor<float>(MBBFreqIndex)[CurrentMBBIndex] = in extractMBBFrequency()
1054 RegallocRunner->getTensor<int64_t>( in extractMBBFrequency()
|
| /external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ |
| D | TimeSeriesDataPointOrBuilder.java | 84 com.google.cloud.aiplatform.v1.TensorboardTensor getTensor(); in getTensor() method
|
| D | TimeSeriesDataPoint.java | 197 public com.google.cloud.aiplatform.v1.TensorboardTensor getTensor() { in getTensor() method in TimeSeriesDataPoint
421 if (!getTensor().equals(other.getTensor())) return false; in equals()
453 hash = (53 * hash) + getTensor().hashCode(); in hashCode()
737 mergeTensor(other.getTensor()); in mergeFrom()
1077 public com.google.cloud.aiplatform.v1.TensorboardTensor getTensor() { in getTensor() method in TimeSeriesDataPoint.Builder
|
| /external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ |
| D | TimeSeriesDataPointOrBuilder.java | 84 com.google.cloud.aiplatform.v1beta1.TensorboardTensor getTensor(); in getTensor() method
|
| D | TimeSeriesDataPoint.java | 197 public com.google.cloud.aiplatform.v1beta1.TensorboardTensor getTensor() { in getTensor() method in TimeSeriesDataPoint
421 if (!getTensor().equals(other.getTensor())) return false; in equals()
453 hash = (53 * hash) + getTensor().hashCode(); in hashCode()
739 mergeTensor(other.getTensor()); in mergeFrom()
1081 public com.google.cloud.aiplatform.v1beta1.TensorboardTensor getTensor() { in getTensor() method in TimeSeriesDataPoint.Builder
|
| /external/pytorch/benchmarks/static_runtime/ |
| D | test_utils.h | 44 at::Tensor getTensor(const at::IValue& ival);
|
| D | test_static_module.cc | 499 auto output_1 = getTensor(mod.forward(inputs)); in TEST()
528 at::Tensor output_1 = getTensor(module.forward(inputs)); in TEST()
534 at::Tensor output_2 = getTensor(output_ivalue); in TEST()
568 at::Tensor output_1 = getTensor(module.forward(args)); in TEST()
579 at::Tensor output_2 = getTensor(output_ivalue); in TEST()
649 auto output_1 = getTensor(mod.forward(inputs)); in TEST()
891 auto output_1 = getTensor(module.forward(inputs)); in TEST()
903 auto output_2 = getTensor(module.forward(inputs)); in TEST()
|
| D | test_utils.cc | 206 at::Tensor getTensor(const at::IValue& ival) { in getTensor() function
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| D | legalize_to_tensor_op.cc | 50 extract.getTensor().getDefiningOp<bufferization::ToTensorOp>(); in matchAndRewrite()
|
| D | lhlo_fuse_linalg.cc | 112 auto alias = toMemref.getTensor(); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | gpu_fusion_rewrite.cc | 292 storeOp.getTensor().getDefiningOp<bufferization::ToTensorOp>(); in annotateLaunchFunc()
336 return isRewritableType(op.getTensor().getType()) && in getRewritableTarget()
|
| D | shape_simplification.cc | 158 auto broadcastOp = op.getTensor().getDefiningOp<BroadcastOp>(); in matchAndRewrite()
|
| /external/pytorch/aten/src/ATen/ |
| D | TensorIterator.h | 113 const Tensor& getTensor() const;
198 return tensor_storage_.getTensor(); in tensor()
208 return original_tensor_storage_.getTensor(); in original_tensor()
|
| D | TensorIterator.cpp | 81 const Tensor& OpaqueOptionalTensorRef::getTensor() const { in getTensor() function in at::internal::OpaqueOptionalTensorRef
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| D | fusion.cc | 45 extract.getTensor().getDefiningOp()); in matchAndRewrite()
|
| D | bufferize_tiled_loop.cc | 158 if (auto toTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>()) { in findExistingSubview()
161 for (Operation *tensorUser : toMemref.getTensor().getUsers()) { in findExistingSubview()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/ |
| D | shape_component_analysis.cc | 579 ShapeOrValueInfo::getValueInfoOf(op.getTensor())); in backwardTensorExtract()
588 auto in = lookup(ShapeOrValueInfo::getValueInfoOf(op.getTensor())); in forwardTensorExtract()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1534 returned_values.push_back(store.getTensor()); in ProcessFusionForConversion()
|