Searched refs:axes_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | reduce_builder.cc | 40 const auto& axes_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
41 if (axes_tensor.allocation_type == kTfLiteMmapRo) { in PopulateSubGraph()
45 graph_builder_->AddConstNodeWithData(tensor_id, axes_tensor); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 1960 const TfLiteTensor& axes_tensor = tensors[node->inputs->data[1]]; in VisitMeanNode() local
1961 TF_LITE_ENSURE_STATUS(CheckTensorType(logging_context, axes_tensor, in VisitMeanNode()
1965 logging_context, axes_tensor, node->inputs->data[1], node_index)); in VisitMeanNode()
1967 logging_context, axes_tensor, node->inputs->data[1], node_index)); in VisitMeanNode()
1969 if (axes_tensor.dims->data[0] != 2) { in VisitMeanNode()
1973 axes_tensor.dims->data[0], node_index); in VisitMeanNode()
1978 reinterpret_cast<const int32_t*>(axes_tensor.data.data); in VisitMeanNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2420 const Tensor* axes_tensor = c->input_tensor(3); in SparseReduceShapeFn() local
2421 if (shape_tensor != nullptr && axes_tensor != nullptr) { in SparseReduceShapeFn()
2423 auto axes_vec = axes_tensor->flat<int32>(); in SparseReduceShapeFn()
|