Home
last modified time | relevance | path

Searched refs:AlgorithmProto (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/compiler/xla/stream_executor/
Ddnn.pb.h55 class AlgorithmProto; variable
71 template<> ::stream_executor::dnn::AlgorithmProto* Arena::CreateMaybeMessage<::stream_executor::dnn…
482 class AlgorithmProto final :
485 inline AlgorithmProto() : AlgorithmProto(nullptr) {} in AlgorithmProto() function
486 ~AlgorithmProto() override;
487 …explicit PROTOBUF_CONSTEXPR AlgorithmProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
489 AlgorithmProto(const AlgorithmProto& from);
490 AlgorithmProto(AlgorithmProto&& from) noexcept in AlgorithmProto() function
491 : AlgorithmProto() { in AlgorithmProto()
495 inline AlgorithmProto& operator=(const AlgorithmProto& from) {
[all …]
Ddnn.pb.cc51 PROTOBUF_CONSTEXPR AlgorithmProto::AlgorithmProto( in AlgorithmProto() function in stream_executor::dnn::AlgorithmProto
64 AlgorithmProto _instance;
162 constexpr AlgorithmProto_MathType AlgorithmProto::DEFAULT_MATH;
163 constexpr AlgorithmProto_MathType AlgorithmProto::TENSOR_OP_MATH;
164 constexpr AlgorithmProto_MathType AlgorithmProto::MathType_MIN;
165 constexpr AlgorithmProto_MathType AlgorithmProto::MathType_MAX;
166 constexpr int AlgorithmProto::MathType_ARRAYSIZE;
917 class AlgorithmProto::_Internal {
919 static const ::PROTOBUF_NAMESPACE_ID::UInt64Value& workspace_size(const AlgorithmProto* msg);
923 AlgorithmProto::_Internal::workspace_size(const AlgorithmProto* msg) { in workspace_size()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dautotuning.proto55 stream_executor.dnn.AlgorithmProto reference_algorithm = 15;
61 // Legacy and unused in new data; superseded by AlgorithmProto.
71 // Legacy and unused in new data; superseded by AlgorithmProto.
85 stream_executor.dnn.AlgorithmProto algorithm = 16;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.proto107 message AlgorithmProto { message
144 AlgorithmProto algorithm = 1;
147 AlgorithmProto algorithm_no_scratch = 2;
H A Ddnn.h786 explicit AlgorithmDesc(AlgorithmProto proto) : proto_(std::move(proto)) {} in AlgorithmDesc()
793 proto_.set_math_type(use_tensor_ops ? AlgorithmProto::TENSOR_OP_MATH in AlgorithmDesc()
794 : AlgorithmProto::DEFAULT_MATH); in AlgorithmDesc()
805 return proto_.math_type() == AlgorithmProto::TENSOR_OP_MATH; in tensor_ops_enabled()
821 AlgorithmProto ToProto() const { return proto_; } in ToProto()
826 AlgorithmProto proto_;
941 const AlgorithmProto& algorithm_proto = algorithm_config_proto.algorithm(); in AlgorithmConfig()
949 const AlgorithmProto& algorithm_no_scratch_proto = in AlgorithmConfig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dbackend_configs.proto26 stream_executor.dnn.AlgorithmProto algorithm = 6;
H A Djitrt_custom_calls.cc778 ? se::dnn::AlgorithmProto::TENSOR_OP_MATH in GetConvDescriptor()
779 : se::dnn::AlgorithmProto::DEFAULT_MATH); in GetConvDescriptor()
H A Dir_emitter_unnested.cc1042 ? se::dnn::AlgorithmProto::TENSOR_OP_MATH in EmitConvolutionThunk()
1043 : se::dnn::AlgorithmProto::DEFAULT_MATH); in EmitConvolutionThunk()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dautotune_serialize.cc41 using stream_executor::dnn::AlgorithmProto;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/
H A Dlhlo_gpu_ops_enums.td60 // These six fields are a TableGen transliteration of AlgorithmProto.
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc992 stream_executor::dnn::AlgorithmProto::TENSOR_OP_MATH, in EmitDnnConvolution()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/turbine/
Dtensorflow_core_proto_java_lite.jarorg/tensorflow/distruntime/ClusterDef$Builder.class <Unknown> package org.tensorflow. ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/javac/
Dtensorflow_core_proto_java_lite.jarMETA-INF/ META-INF/MANIFEST.MF com/ com/google/ com/ ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/combined/
DFederatedCompute.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx/ META- ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/jarjar/
DFederatedCompute.jarxla/Protocol.class Protocol.java package xla public final xla.Protocol extends java.lang ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/OnDevicePersonalization/android_common_apex33/withres/
DOnDevicePersonalization.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx.annotation_annotation- ...