Searched defs:perm_shape (Results 1 – 8 of 8) sorted by relevance
134 const std::array<int32_t, 1> perm_shape{ in CreateTfLiteModel() local
252 const TensorShape& perm_shape = permutation_indices.shape(); in ValidateInputs() local
26 std::initializer_list<int> perm_shape, in TransposeOpModel()
238 std::initializer_list<int> perm_shape, in TransposeOpConstModel()259 std::initializer_list<int> perm_shape) { in TransposeOpDynamicModel()
584 ShapeHandle perm_shape; in __anon6adce4611002() local
124 ShapeHandle perm_shape = c->input(1); in TransposeShapeFn() local
57 …arm_compute::PermutationVector perm_shape = (data_layout == arm_compute::DataLayout::NCHW) ? arm_c… in compute_permutation_parameters() local
1624 std::initializer_list<int> perm_shape, in TransposeSimpleModel()