| /aosp_15_r20/external/XNNPACK/src/subgraph/ |
| H A D | square-root.c | 38 const size_t num_input_dims = values[input_id].shape.num_dims; in create_square_root_operator() local
|
| H A D | abs.c | 35 const size_t num_input_dims = values[input_id].shape.num_dims; in create_abs_operator() local
|
| H A D | ceiling.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_ceiling_operator() local
|
| H A D | softmax.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_softmax_operator() local
|
| H A D | bankers-rounding.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_bankers_rounding_operator() local
|
| H A D | square.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_square_operator() local
|
| H A D | hardswish.c | 38 const size_t num_input_dims = values[input_id].shape.num_dims; in create_hardswish_operator() local
|
| H A D | floor.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_floor_operator() local
|
| H A D | negate.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_negate_operator() local
|
| H A D | sigmoid.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_sigmoid_operator() local
|
| H A D | elu.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_elu_operator() local
|
| H A D | prelu.c | 39 const size_t num_input_dims = values[input_id].shape.num_dims; in create_prelu_operator() local
|
| H A D | leaky-relu.c | 36 const size_t num_input_dims = values[input_id].shape.num_dims; in create_leaky_relu_operator() local
|
| H A D | clamp.c | 37 const size_t num_input_dims = values[input_id].shape.num_dims; in create_clamp_operator() local
|
| H A D | convert.c | 37 const size_t num_input_dims = values[input_id].shape.num_dims; in create_convert_operator() local
|
| H A D | global-average-pooling.c | 37 const size_t num_input_dims = values[input_id].shape.num_dims; in create_global_average_pooling_operator() local
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | fully-connected.cc | 46 size_t num_input_dims = shape_dist(rng); in FullyConnectedTestBase() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | batch_matmul.cc | 456 const int32_t num_input_dims = input_shape.DimensionsCount(); in EvalHybrid() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | prepare_tf.cc | 714 int num_input_dims = ranked_input_type.getRank(); in matchAndRewrite() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 1294 def _test_pixel_shuffle_unshuffle_helper(num_input_dims, valid_channels_dim=True, argument
1354 …def _test_pixel_unshuffle_error_case_helper(num_input_dims, valid_height_dim=True, valid_width_dim… argument
1374 def _test_pixel_shuffle_unshuffle_for_input_dims(num_input_dims): argument
|
| H A D | test_nn.py | 4400 def _test_pixel_shuffle_unshuffle_helper(num_input_dims, valid_channels_dim=True, argument
4442 …def _test_pixel_unshuffle_error_case_helper(num_input_dims, valid_height_dim=True, valid_width_dim… argument
4462 def _test_pixel_shuffle_unshuffle_for_input_dims(num_input_dims): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer.cc | 1780 const int num_input_dims = context->src_format.length(); in UpdateSqueezeDims() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 3184 int num_input_dims = ranked_input_type.getRank(); in verify() local
|