Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h193 struct MeanAttributes { struct
507 BHWC CalculateOutputShape(const BHWC& input, const MeanAttributes& attr);
510 BHWDC CalculateOutputShape(const BHWDC& input, const MeanAttributes& attr);
Doperations.cc584 BHWC CalculateOutputShape(const BHWC& input, const MeanAttributes& attr) { in CalculateOutputShape()
592 BHWDC CalculateOutputShape(const BHWDC& input, const MeanAttributes& attr) { in CalculateOutputShape()
Dmodel_builder.cc2289 MeanAttributes attr; in Parse()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dmean_test.cc42 MeanAttributes attr; in TEST()
Dmean.cc178 const auto& attr = absl::any_cast<const MeanAttributes&>(ctx.op_attr); in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dglobal_pooling_to_reduce_op.cc65 MeanAttributes mean_attr; in ApplyToNode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc430 auto attr = absl::any_cast<MeanAttributes>(node.operation.attributes); in GPUOperationFromNode()