Searched refs:input_shape_vector (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/python/pyarmnn/test/ |
| H A D | test_descriptors.py | 494 input_shape_vector = [ann.TensorShape((2, 1)), ann.TensorShape((3, 1)), ann.TensorShape((4, 1))]
495 desc = ann.CreateDescriptorForConcatenation(input_shape_vector, 0)
504 input_shape_vector = [ann.TensorShape((1, 2)), ann.TensorShape((3, 4)), ann.TensorShape((5, 6))]
506 desc = ann.CreateDescriptorForConcatenation(input_shape_vector, 0)
515 def test_createdescriptorforconcatenation_rubbish_assignment_shape_vector(input_shape_vector): argument
517 desc = ann.CreateDescriptorForConcatenation(input_shape_vector, 0)
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_concat_op.cc | 134 auto input_shape_vector = shapes[i].vec<int64_t>(); in Compute() local
135 for (int j = 0; j < input_shape_vector.size(); j++) { in Compute()
137 MultiplyWithoutOverflow(new_num_elements, input_shape_vector(j)); in Compute()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 452 std::vector<c10::ShapeSymbol> input_shape_vector = in ComputeShapeFromReshape() local
456 !input_shape_vector.empty() || !shape_vector.empty(), in ComputeShapeFromReshape()
461 if (input_shape_vector.empty()) { in ComputeShapeFromReshape()
492 for (const c10::ShapeSymbol& input_shape : input_shape_vector) { in ComputeShapeFromReshape()
515 target_shape = input_shape_vector[i]; in ComputeShapeFromReshape()
543 shape_vector[i].value() == 0 ? input_shape_vector[i] : shape_vector[i]); in ComputeShapeFromReshape()
552 shape_vector[i].value() == 0 ? input_shape_vector[i] : shape_vector[i]); in ComputeShapeFromReshape()
|