Home
last modified time | relevance | path

Searched defs:output_type (Results 1 – 25 of 232) sorted by relevance

12345678910

/external/tensorflow/tensorflow/python/data/kernel_tests/
Drange_test.py37 def testStop(self, output_type): argument
50 def testStartStop(self, output_type): argument
63 def testStartStopStep(self, output_type): argument
78 def testZeroStep(self, output_type): argument
91 def testNegativeStep(self, output_type): argument
106 def testStopLessThanStart(self, output_type): argument
119 def testStopLessThanStartWithPositiveStep(self, output_type): argument
134 def testStopLessThanStartWithNegativeStep(self, output_type): argument
200 def testMultipleCombinations(self, start, stop, step, output_type): argument
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_gpu_op.h26 #define REGISTER_ALIASED_GPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
31 #define REGISTER_GPU_KERNEL(tf_op, input_type, output_type) \ argument
35 #define REGISTER_COMPLEX_GPU_KERNEL(tf_op, input_type, output_type) \ argument
48 output_type) \ argument
53 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
65 #define GENERATE_UNARY_GPU_KERNEL2(tf_op, input_type, output_type) \ argument
68 #define GENERATE_UNARY_GPU_KERNEL3(tf_op, input_type, output_type, \ argument
80 output_type) \ argument
85 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
97 #define GENERATE_BINARY_GPU_KERNEL2(tf_op, input_type, output_type) \ argument
[all …]
Dbase_op.h140 #define MLIR_FUNCTION(tf_op, platform, input_type, output_type) \ argument
143 #define MLIR_OP(tf_op, platform, input_type, output_type) \ argument
149 output_type, additional_cstrs) \ argument
157 #define REGISTER_KERNEL(tf_op, platform, input_type, output_type, \ argument
162 #define REGISTER_COMPLEX_KERNEL(tf_op, platform, input_type, output_type) \ argument
182 output_type, additional_cstrs) \ argument
187 tf_op, platform, input_type, output_type, casted_input_type, \ argument
202 #define GENERATE_UNARY_KERNEL2(tf_op, platform, input_type, output_type) \ argument
206 #define GENERATE_UNARY_KERNEL3(tf_op, platform, input_type, output_type, \ argument
238 output_type, additional_cstrs) \ argument
[all …]
Dgpu_op_cast.cc37 #define GENERATE_AND_REGISTER_CAST_GPU(input_type, output_type) \ argument
/external/gemmlowp/meta/generators/
Dgemm_NxMxK_neon.py173 def BuildName(output_type, aligned, rows, cols, leftover): argument
187 def GetGemmParameters(output_type, extra_params=None): argument
205 def GetStridedGemmParameters(output_type): argument
209 def GenerateGemm(emitter, output_type, aligned, rows, cols, leftovers): argument
237 def GenerateGemmCall(emitter, output_type, aligned, m_mod, n_mod, leftovers): argument
244 def GenerateGemmSwitch3(emitter, output_type, aligned, m_mod, n_mod): argument
258 def GenerateGemmSwitch2(emitter, output_type, aligned, m_mod): argument
272 def GenerateGemmSwitch1(emitter, output_type, aligned): argument
286 def BuildMainGemmName(output_type): argument
297 def BuildStridedMainGemmName(output_type): argument
[all …]
Dgemv_1xMxK_neon.py140 def BuildName(output_type, aligned, cols, leftover): argument
154 def GetGemvParameters(output_type): argument
170 def GenerateGemv(emitter, output_type, aligned, cols, leftovers): argument
197 def GenerateGemvCall(emitter, output_type, aligned, m_mod, leftovers): argument
204 def GenerateGemvSwitch2(emitter, output_type, aligned, n_mod): argument
218 def GenerateGemvSwitch1(emitter, output_type, aligned): argument
232 def BuildMainGemvName(output_type): argument
243 def GenerateMainGemvFunction(emitter, output_type): argument
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc147 TensorType output_type = in matchAndRewrite() local
165 TensorType output_type = in matchAndRewrite() local
181 TensorType output_type = in matchAndRewrite() local
195 TensorType output_type = in matchAndRewrite() local
216 TensorType output_type = in matchAndRewrite() local
230 TensorType output_type = in matchAndRewrite() local
245 TensorType output_type = in matchAndRewrite() local
259 TensorType output_type = in matchAndRewrite() local
274 TensorType output_type = in matchAndRewrite() local
299 TensorType output_type = in matchAndRewrite() local
[all …]
Dlegalize_tfl.cc182 ShapedType output_type = in matchAndRewrite() local
232 ShapedType output_type = in matchAndRewrite() local
287 ShapedType output_type = in matchAndRewrite() local
344 ShapedType output_type = result.getType().dyn_cast<ShapedType>(); in prepareMatchAndRewriteComparison() local
482 ShapedType output_type = in matchAndRewriteAddSub() local
670 ShapedType output_type = in matchAndRewrite() local
713 ShapedType output_type = in matchAndRewrite() local
766 ShapedType output_type = in matchAndRewrite() local
846 ShapedType output_type = in matchAndRewrite() local
873 ShapedType output_type = in matchAndRewrite() local
[all …]
/external/compiler-rt/test/asan/android_commands/
Dandroid_compile.py11 output_type = 'executable' variable
17 output_type = 'shared' variable
19 output_type = 'object' variable
/external/perfetto/src/ipc/protoc_plugin/
Dipc_plugin.cc132 std::string output_type(method.output_type()->name()); in ForEachMethod() local
178 const std::string& output_type) { in GenerateServiceHeader()
194 const std::string& output_type) { in GenerateServiceHeader()
225 const std::string& output_type) { in GenerateServiceCpp()
238 const std::string& output_type) { in GenerateServiceCpp()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc80 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall()
122 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
145 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitMathCall()
182 PrimitiveType output_type = op->shape().element_type(); in EmitFloatBinaryOp() local
212 PrimitiveType output_type = op->shape().element_type(); in EmitPowerOp() local
/external/rust/android-crates-io/crates/protobuf/src/reflect/service/
Dindex.rs28 pub(crate) output_type: ForwardProtobufTypeBox, field
37 let output_type = building.resolve_message(proto.output_type())?; in index() localVariable
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Darithmetic_count_util.h29 auto output_type = in GetFirstOutputCount() local
61 auto output_type = in GetArithmeticCountForConvAndFullyconnectedOp() local
/external/tensorflow/tensorflow/c/kernels/ops/
Dbitcast.cc26 TF_DataType output_type, TF_Status* status) { in ComputeNewShape()
82 TF_DataType output_type; in bitcast_shape_inference_fn() local
/external/tensorflow/tensorflow/core/profiler/
Dtfprof_options.cc36 tensorflow::Status ParseOutput(const string& output_opt, string* output_type, in ParseOutput()
133 string output_type; in FromProtoStr() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc102 auto output_type = output.getType().dyn_cast_or_null<RankedTensorType>(); in EnsureBias() local
300 auto output_type = output.getType().cast<RankedTensorType>(); in matchAndRewrite() local
385 auto output_type = slice_op.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
483 auto output_type = output.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
539 auto output_type = concat_op.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
619 auto output_type = output.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
678 auto output_type = output.getType().dyn_cast_or_null<ShapedType>(); in matchAndRewrite() local
/external/zopfli/src/zopfli/
Dzopfli_bin.c100 ZopfliFormat output_type, in CompressFile()
146 ZopfliFormat output_type = ZOPFLI_FORMAT_GZIP; in main() local
Dzopfli_lib.c28 void ZopfliCompress(const ZopfliOptions* options, ZopfliFormat output_type, in ZopfliCompress()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibrator.py145 output_type, argument
187 output_type, argument
/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/segmenter/
Dimage_segmenter_jni.cc62 jint output_type, in ConvertToProtoOptions()
189 jstring display_names_locale, jint output_type, jint num_threads) { in Java_org_tensorflow_lite_task_vision_segmenter_ImageSegmenter_initJniWithModelFdAndOptions()
207 jstring display_names_locale, jint output_type, jint num_threads) { in Java_org_tensorflow_lite_task_vision_segmenter_ImageSegmenter_initJniWithByteBuffer()
/external/sdv/vsomeip/third_party/boost/date_time/include/boost/date_time/
Dstring_convert.hpp24 typedef std::basic_string<OutputT> output_type; in convert_string_type() typedef
/external/ethtool/
Djson_print.h25 enum output_type { enum
/external/tensorflow/tensorflow/dtensor/mlir/ir/
Dtf_dtensor.cc118 RankedTensorType output_type = in verify() local
174 RankedTensorType output_type = in verify() local
/external/rust/android-crates-io/crates/diplomat_core/src/ast/
Dtraits.rs28 pub output_type: Option<TypeName>, field
83 let output_type = match &fct.sig.output { in new() localVariable
/external/tensorflow/tensorflow/lite/tools/optimize/python/
Dmodify_model_interface.cc32 const int input_type, const int output_type) -> int { in PYBIND11_MODULE()

12345678910