Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/python/data/kernel_tests/
Drange_test.py40 def testStop(self, output_type): argument
53 def testStartStop(self, output_type): argument
66 def testStartStopStep(self, output_type): argument
81 def testZeroStep(self, output_type): argument
94 def testNegativeStep(self, output_type): argument
109 def testStopLessThanStart(self, output_type): argument
122 def testStopLessThanStartWithPositiveStep(self, output_type): argument
137 def testStopLessThanStartWithNegativeStep(self, output_type): argument
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_cpu_op.h27 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
34 #define GENERATE_UNARY_CPU_KERNEL2(tf_op, input_type, output_type) \ argument
37 #define GENERATE_UNARY_CPU_KERNEL3(tf_op, input_type, output_type, \ argument
42 #define REGISTER_ALIASED_CPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
45 #define REGISTER_CPU_KERNEL(tf_op, input_type, output_type) \ argument
48 #define REGISTER_COMPLEX_CPU_KERNEL(tf_op, input_type, output_type) \ argument
58 output_type) \ argument
62 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
69 #define GENERATE_BINARY_CPU_KERNEL2(tf_op, input_type, output_type) \ argument
Dbase_gpu_op.h27 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
34 #define GENERATE_UNARY_GPU_KERNEL2(tf_op, input_type, output_type) \ argument
37 #define GENERATE_UNARY_GPU_KERNEL3(tf_op, input_type, output_type, \ argument
42 #define REGISTER_ALIASED_GPU_KERNEL(tf_op, mlir_op, input_type, output_type) \ argument
45 #define REGISTER_GPU_KERNEL(tf_op, input_type, output_type) \ argument
48 #define REGISTER_COMPLEX_GPU_KERNEL(tf_op, input_type, output_type) \ argument
58 output_type) \ argument
62 tf_op, input_type, output_type, casted_input_type, casted_output_type) \ argument
69 #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
189 output_type, casted_input_type, \ argument
198 #define GENERATE_BINARY_KERNEL2(tf_op, platform, input_type, output_type) \ argument
202 #define GENERATE_BINARY_KERNEL3(tf_op, platform, input_type, output_type, \ argument
278 output_type, casted_input_type, \ argument
[all …]
Dgpu_op_cast.cc35 #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 RankedTensorType output_type = in matchAndRewrite() local
164 RankedTensorType output_type = in matchAndRewrite() local
180 RankedTensorType output_type = in matchAndRewrite() local
194 RankedTensorType output_type = in matchAndRewrite() local
214 RankedTensorType output_type = in matchAndRewrite() local
228 RankedTensorType output_type = in matchAndRewrite() local
242 RankedTensorType output_type = in matchAndRewrite() local
256 RankedTensorType output_type = in matchAndRewrite() local
271 RankedTensorType output_type = in matchAndRewrite() local
296 RankedTensorType output_type = in matchAndRewrite() local
[all …]
Dlegalize_tfl.cc176 ShapedType output_type = in matchAndRewrite() local
226 ShapedType output_type = in matchAndRewrite() local
283 ShapedType output_type = result.getType().dyn_cast<ShapedType>(); in prepareMatchAndRewriteComparison() local
426 ShapedType output_type = in matchAndRewriteAddSub() local
618 ShapedType output_type = in matchAndRewrite() local
664 ShapedType output_type = in matchAndRewrite() local
717 ShapedType output_type = in matchAndRewrite() local
799 ShapedType output_type = in matchAndRewrite() local
827 ShapedType output_type = in matchAndRewrite() local
882 ShapedType output_type = in matchAndRewrite() local
[all …]
Dlegalize_common.cc453 ShapedType output_type = output_val.getType().dyn_cast<ShapedType>(); in convertMultiplyOp() local
1148 RankedTensorType output_type = in convertExpandDimsOp() local
1201 RankedTensorType output_type = in convertSqueezeOp() local
1268 RankedTensorType output_type = in convertEluOp() local
1318 RankedTensorType output_type = in convertSoftmaxOp() local
1685 RankedTensorType output_type = in convertLogSoftmaxOp() local
1751 RankedTensorType output_type = in convertSpaceToDepthOp() local
1843 RankedTensorType output_type = in convertDepthToSpaceOp() local
2248 RankedTensorType output_type = in convertFloorDivOp() local
2284 RankedTensorType output_type = in convertFloorModOp() 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.cc82 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitDeviceMathCall()
124 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type) { in EmitLlvmIntrinsicMathCall()
147 absl::Span<const PrimitiveType> input_types, PrimitiveType output_type, in EmitMathCall()
184 PrimitiveType output_type = op->shape().element_type(); in EmitFloatBinaryOp() local
214 PrimitiveType output_type = op->shape().element_type(); in EmitPowerOp() 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/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/lite/python/optimize/
Dcalibrator.py85 output_type, argument
133 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.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/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/mesa3d/src/compiler/nir/
Dnir_opcodes.py35 def __init__(self, name, output_size, output_type, input_sizes, argument
153 def opcode(name, output_size, output_type, input_sizes, input_types, argument
166 def unop_horiz(name, output_size, output_type, input_size, input_type, argument
171 def unop_reduce(name, output_size, output_type, input_type, prereduce_expr, argument
543 def binop_reduce(name, output_size, output_type, src_type, prereduce_expr, argument
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc98 auto output_type = output.getType().dyn_cast_or_null<RankedTensorType>(); in EnsureBias() local
296 auto output_type = output.getType().cast<RankedTensorType>(); in matchAndRewrite() local
381 auto output_type = slice_op.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
479 auto output_type = output.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
535 auto output_type = concat_op.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
/external/rust/crates/grpcio-sys/grpc/src/compiler/
Dphp_generator.cc69 const Descriptor* output_type = method->output_type(); in PrintMethod() local
137 const Descriptor* output_type = method->output_type(); in PrintServerMethod() local
/external/iproute2/include/
Djson_print.h26 enum output_type { enum

12345678