Searched defs:shapeOut (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Broadcast.cpp | 82 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16()
97 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32()
124 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16()
131 int32_t activation, T* out, const Shape& shapeOut) { in addQuant8()
236 int32_t activation, float* out, const Shape& shapeOut) { in mulFloat32()
263 int32_t activation, _Float16* out, const Shape& shapeOut) { in mulFloat16()
270 int32_t activation, T* out, const Shape& shapeOut) { in mulQuant8()
317 int32_t activation, float* out, const Shape& shapeOut) { in subFloat32()
334 int32_t activation, _Float16* out, const Shape& shapeOut) { in subFloat16()
341 int32_t activation, T* out, const Shape& shapeOut) { in subQuant8()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| H A D | symbolic_shape_optimization.cc | 146 auto shapeOut = analysis.GetValueInfo(shape); in analyzeDynamicBroadcastInDimExpandingBehavior() local
|