Home
last modified time | relevance | path

Searched defs:axis_data (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_gather_attributes.cc46 const auto& axis_data = indices_array.GetBuffer<ArrayDataType::kInt32>().data; in Run() local
Dresolve_tensorflow_concat.cc64 const auto& axis_data = axis_array.GetBuffer<ArrayDataType::kInt32>().data; in Run() local
/external/ComputeLibrary/tests/validation/fixtures/
DReverseFixture.h97 auto axis_data = AccessorType(axis); in compute_target() local
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dpooling_layer_builder.cc151 const int* axis_data = GetTensorData<int>(axis); in IsMeanOpSupported() local
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFFTConvolutionLayer.cpp257 auto axis_data = reinterpret_cast<uint32_t *>(_flip_axis.buffer()); in configure() local
DNEDeconvolutionLayer.cpp191 auto axis_data = reinterpret_cast<uint32_t *>(_flip_axis.buffer()); in configure() local
/external/ComputeLibrary/src/runtime/CL/functions/
DCLFFTConvolutionLayer.cpp263 auto axis_data = reinterpret_cast<uint32_t *>(_flip_axis.buffer()); in configure() local
DCLDirectDeconvolutionLayer.cpp188 auto axis_data = reinterpret_cast<uint32_t *>(_flip_axis.buffer()); in configure() local
/external/tensorflow/tensorflow/lite/kernels/
Droll.cc202 std::vector<int32_t> axis_data = ExtractIntegerVector(axis); in Eval() local
Dreduce.cc337 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis()