Home
last modified time | relevance | path

Searched defs:shapeDims (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp338 std::vector<unsigned int> shapeDims; in ToTensorInfo() local
349 std::vector<unsigned int> shapeDims; in ToTensorInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dsymbolic_shape_optimization.cc743 auto shapeDims = analysis.GetValueInfo(shape); in isKnownBroadcastable() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/
H A Dshape_component_analysis.cc338 auto shapeDims = in forwardDynamicReshapeShape() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp1680 uint32_t shapeDims[1] = {2}; in reshapeOpTest() local