Home
last modified time | relevance | path

Searched refs:elementwise (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/lite/micro/kernels/
Delementwise.cc27 namespace elementwise { namespace
120 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_ABS()
121 /*invoke=*/elementwise::AbsEval, in Register_ABS()
132 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_SIN()
133 /*invoke=*/elementwise::SinEval, in Register_SIN()
144 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_COS()
145 /*invoke=*/elementwise::CosEval, in Register_COS()
156 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_LOG()
157 /*invoke=*/elementwise::LogEval, in Register_LOG()
168 elementwise::GenericPrepare<elementwise::IsNumericSupportedType>, in Register_SQRT()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Delementwise.cc33 namespace elementwise { namespace
300 elementwise::ElementWiseQuantizedInit, in Register_ABS()
301 elementwise::ElementWiseQuantizedFree, in Register_ABS()
302 elementwise::GenericPrepare<elementwise::IsAbsSupportedType, in Register_ABS()
303 elementwise::kAbsName>, in Register_ABS()
304 elementwise::AbsEval}; in Register_ABS()
311 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_SIN()
312 elementwise::kSinName>, in Register_SIN()
313 elementwise::SinEval}; in Register_SIN()
320 elementwise::GenericPrepare<elementwise::IsNumericSupportedType, in Register_COS()
[all …]
DAndroid.bp86 "elementwise.cc",
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Xdivy.pbtxt3 summary: "Returns 0 if x == 0, and x / y otherwise, elementwise."
Dapi_def_Xlogy.pbtxt3 summary: "Returns 0 if x == 0, and x * log(y) otherwise, elementwise."
Dapi_def_Xlog1py.pbtxt3 summary: "Returns 0 if x == 0, and x * log1p(y) otherwise, elementwise."
Dapi_def_Complex.pbtxt7 operation returns complex numbers elementwise of the form \\(a + bj\\), where
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Maximum.pbtxt24 The reduction version of this elementwise operation is `tf.math.reduce_max`
Dapi_def_Minimum.pbtxt27 The reduction version of this elementwise operation is `tf.math.reduce_min`
Dapi_def_Add.pbtxt66 The reduction version of this elementwise operation is `tf.math.reduce_sum`
Dapi_def_LogicalOr.pbtxt55 The reduction version of this elementwise operation is `tf.math.reduce_any`.
Dapi_def_LogicalAnd.pbtxt55 The reduction version of this elementwise operation is `tf.math.reduce_all`.
/external/llvm-project/mlir/integration_test/Dialect/Linalg/CPU/
Dtest-elementwise.mlir1 // RUN: mlir-opt %s -convert-elementwise-to-linalg -std-bufferize -tensor-constant-bufferize -linal…
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
DBUILD199 name = "elementwise",
200 srcs = ["elementwise.cc"],
201 hdrs = ["elementwise.h"],
221 ":elementwise",
753 "elementwise",
/external/llvm-project/mlir/test/Dialect/Linalg/
Dconvert-elementwise-to-linalg.mlir1 // RUN: mlir-opt -convert-elementwise-to-linalg -split-input-file %s | FileCheck %s
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dchlo_ops.td64 // CHLO binary elementwise op definitions.
292 // XLA binary logical elementwise op definitions.
293 // The same description as the arithmetic binary elementwise ops applies.
657 Compares `lhs` and `rhs` elementwise according to `comparison_direction`
Dinfer_fusibility_op_interface.td108 For example, the effective workload shape of an elementwise op is its
/external/tensorflow/tensorflow/lite/micro/examples/micro_speech/apollo3/
DREADME.md39 * cmsis_windowed_input.txt: the sinusoid after multiplying elementwise
57 * micro_windowed_input.txt: the sinusoid after multiplying elementwise
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlower_complex_patterns.td27 // Add and subtraction are elementwise and can be distributed across the real
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dserialization_base.fbs233 elementwise:bool;
Dserialization_base_generated.h2148 bool elementwise() const { return GetField<uint8_t>(VT_ELEMENTWISE, 0) != 0; } in elementwise() function
2246 void add_elementwise(bool elementwise) { in add_elementwise()
2248 static_cast<uint8_t>(elementwise), 0); in add_elementwise()
2319 bool elementwise = false, bool linkable = false,
2350 builder_.add_elementwise(elementwise);
2364 bool elementwise = false, bool linkable = false,
2397 tensor_to_grid, elementwise, linkable, check_src_channels_size,
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/
DPasses.td14 def ConvertElementwiseToLinalg : FunctionPass<"convert-elementwise-to-linalg"> {
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp68 "kernels/elementwise.cc",
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
DBUILD103 "//tensorflow/lite/delegates/gpu/common/tasks:elementwise",
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
DBUILD430 name = "elementwise",
431 srcs = ["elementwise.cc"],
432 hdrs = ["elementwise.h"],
448 ":elementwise",

123