Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/python/data/kernel_tests/
Drange_test.py39 def testStop(self, output_type): argument
52 def testStartStop(self, output_type): argument
65 def testStartStopStep(self, output_type): argument
80 def testZeroStep(self, output_type): argument
93 def testNegativeStep(self, output_type): argument
108 def testStopLessThanStart(self, output_type): argument
121 def testStopLessThanStartWithPositiveStep(self, output_type): argument
136 def testStopLessThanStartWithNegativeStep(self, output_type): argument
/external/llvm-project/compiler-rt/test/sanitizer_common/ios_commands/
Diossim_compile.py6 output_type = 'executable' variable
12 output_type = 'shared' variable
14 output_type = 'dylib' variable
16 output_type = 'object' variable
18 output_type = 'assembly' variable
/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/core/kernels/mlir_generated/
Dbase_cpu_op.h29 #define GENERATE_UNARY_CPU_KERNEL2(tf_op, input_type, output_type) \ argument
32 #define REGISTER_ALIASED_CPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
35 #define REGISTER_CPU_KERNEL(tf_op, input_type, output_type) \ argument
38 #define REGISTER_COMPLEX_CPU_KERNEL(tf_op, input_type, output_type) \ argument
48 output_type) \ argument
54 #define GENERATE_BINARY_CPU_KERNEL2(tf_op, input_type, output_type) \ argument
Dbase_gpu_op.h29 #define GENERATE_UNARY_GPU_KERNEL2(tf_op, input_type, output_type) \ argument
32 #define REGISTER_ALIASED_GPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
35 #define REGISTER_GPU_KERNEL(tf_op, input_type, output_type) \ argument
38 #define REGISTER_COMPLEX_GPU_KERNEL(tf_op, input_type, output_type) \ argument
48 output_type) \ argument
54 #define GENERATE_BINARY_GPU_KERNEL2(tf_op, input_type, output_type) \ argument
Dbase_op.h146 #define MLIR_FUNCTION(tf_op, platform, input_type, output_type) \ argument
149 #define MLIR_OP(tf_op, platform, input_type, output_type) \ argument
153 output_type) \ argument
160 #define REGISTER_KERNEL(tf_op, platform, input_type, output_type) \ argument
163 #define REGISTER_COMPLEX_KERNEL(tf_op, platform, input_type, output_type) \ argument
184 output_type) \ argument
191 #define GENERATE_BINARY_KERNEL2(tf_op, platform, input_type, output_type) \ argument
227 #define GENERATE_UNARY_KERNEL2(tf_op, platform, input_type, output_type) \ argument
Dgpu_op_cast.cc31 #define GENERATE_AND_REGISTER_CAST_GPU(input_type, output_type) \ argument
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc141 RankedTensorType output_type = in matchAndRewrite() local
163 RankedTensorType output_type = in matchAndRewrite() local
184 RankedTensorType output_type = in matchAndRewrite() local
198 RankedTensorType output_type = in matchAndRewrite() local
218 RankedTensorType output_type = in matchAndRewrite() local
232 RankedTensorType output_type = in matchAndRewrite() local
246 RankedTensorType output_type = in matchAndRewrite() local
260 RankedTensorType output_type = in matchAndRewrite() local
275 RankedTensorType output_type = in matchAndRewrite() local
300 RankedTensorType output_type = in matchAndRewrite() local
[all …]
Dlegalize_tfl.cc149 RankedTensorType output_type = in matchAndRewrite() local
206 RankedTensorType output_type = in matchAndRewrite() local
266 RankedTensorType output_type = in matchAndRewrite() local
328 RankedTensorType output_type = in matchAndRewrite() local
397 RankedTensorType output_type = in matchAndRewrite() local
461 RankedTensorType output_type = in matchAndRewrite() local
527 RankedTensorType output_type = in matchAndRewrite() local
625 RankedTensorType output_type = in matchAndRewrite() local
798 RankedTensorType output_type = in matchAndRewrite() local
834 RankedTensorType output_type = in matchAndRewrite() local
[all …]
Dlegalize_common.cc448 RankedTensorType output_type = in convertMultiplyOp() local
1242 RankedTensorType output_type = in convertExpandDimsOp() local
1295 RankedTensorType output_type = in convertSqueezeOp() local
1362 RankedTensorType output_type = in convertEluOp() local
1411 RankedTensorType output_type = in convertSoftmaxOp() local
1614 RankedTensorType output_type = in convertLogSoftmaxOp() local
1680 RankedTensorType output_type = in convertSpaceToDepthOp() local
1770 RankedTensorType output_type = in convertDepthToSpaceOp() local
2156 RankedTensorType output_type = in convertFloorDivOp() local
2184 RankedTensorType output_type = in convertFloorModOp() local
[all …]
Dlegalize_utils.cc30 RankedTensorType output_type, Value input_val, double scale, in buildRescale()
56 auto output_type = in buildRescaleToInt32() local
65 RankedTensorType output_type, Value input_val, in buildRescaleFromInt32()
82 RankedTensorType output_type) { in buildRescaleOpConvOutput()
306 RankedTensorType filter_type, RankedTensorType output_type, in getTransposeConv2dPaddingValues()
412 auto output_type = RankedTensorType::get( in getUnquantizedBias() local
/external/llvm-project/compiler-rt/test/sanitizer_common/android_commands/
Dandroid_compile.py11 output_type = 'executable' variable
17 output_type = 'shared' variable
19 output_type = 'object' variable
/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.cc81 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall()
123 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
146 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitMathCall()
166 PrimitiveType output_type = op->shape().element_type(); in EmitFloatBinaryOp() local
196 PrimitiveType output_type = op->shape().element_type(); in EmitPowerOp() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/estimators/
Darithmetic_count_util.h27 auto output_type = in GetArithmeticCountForBroadcastableOp() local
59 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/compiler/xla/client/lib/
Darithmetic.cc143 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min, in ArgMinMax()
175 XlaOp ArgMinMaxTwoPass(XlaOp input, PrimitiveType output_type, int axis, in ArgMinMaxTwoPass()
220 XlaOp ArgMax(XlaOp input, PrimitiveType output_type, int axis, bool stable, in ArgMax()
225 XlaOp ArgMin(XlaOp input, PrimitiveType output_type, int axis, bool stable, in ArgMin()
230 XlaOp ArgMaxTwoPass(XlaOp input, PrimitiveType output_type, int axis, in ArgMaxTwoPass()
235 XlaOp ArgMinTwoPass(XlaOp input, PrimitiveType output_type, int axis, in ArgMinTwoPass()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibrator.py67 output_type, argument
109 output_type, argument
/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/zopfli/src/zopfli/
Dzopfli_bin.c83 ZopfliFormat output_type, in CompressFile()
130 ZopfliFormat output_type = ZOPFLI_FORMAT_GZIP; in main() local
Dzopfli_lib.c28 void ZopfliCompress(const ZopfliOptions* options, ZopfliFormat output_type, in ZopfliCompress()
/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/iproute2/include/
Djson_print.h26 enum output_type { enum

1234567