Home
last modified time | relevance | path

Searched refs:REDUCE_PRODUCT (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dreduce.cc61 } else if (op_type == OperationType::REDUCE_PRODUCT) { in MakeOp()
291 } else if (op_type == OperationType::REDUCE_PRODUCT) { in GetReduceKernelCode()
319 if (op_type == OperationType::REDUCE_PRODUCT || in GetReduceKernelCode()
341 } else if (op_type == OperationType::REDUCE_PRODUCT) { in GetReduceKernelCode()
394 } else if (op_type == OperationType::REDUCE_PRODUCT) { in GetReduceKernelCode()
Dreduce_test_util.cc98 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_PRODUCT, in ReduceProductChannelsTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h76 REDUCE_PRODUCT, enumerator
Doperations.cc155 case OperationType::REDUCE_PRODUCT: in ToString()
240 {"reduce_product", OperationType::REDUCE_PRODUCT}, in OperationTypeFromString()
Dmodel_builder.cc2389 OperationType::REDUCE_PRODUCT); in NewOperationParser()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc555 case OperationType::REDUCE_PRODUCT: in GPUOperationFromNode()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md2367 %output = lower_reduce_op<tosa.REDUCE_PRODUCT>(%input, %output.shape, %reduction_indicies, keep_dim…