Searched defs:w_shape (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | BatchNormalizationLayerFusionFixture.h | 49 …void setup(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShape dst_shape,… in setup()
83 …TensorType compute_target(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorS… in compute_target()
159 …SimpleTensor<T> compute_reference(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape,… in compute_reference()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmDirectConv2d.cpp | 169 const TensorShape w_shape = weights->tensor_shape(); in validate() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_constants.cc | 240 const auto& w_shape = attr.weights.shape; in IsConvConstantsSupported() local
|
| /external/pytorch/torch/utils/ |
| D | flop_counter.py | 132 def conv_flop(x_shape, w_shape, _bias, _stride, _padding, _dilation, transposed, *args, out_shape=N… argument
141 w_shape, argument
|
| /external/libxaac/encoder/ |
| D | ixheaace_cplx_pred.c | 175 FLOAT64 *ptr_dmx_im, WORD32 window, const WORD32 w_shape, in iusace_estimate_dmx_im()
|
| /external/libxaac/decoder/ |
| D | ixheaacd_ext_ch_ele.c | 428 WORD32 window, const WORD32 w_shape, in ixheaacd_estimate_dmx_im()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 6972 auto w_shape = llvm::to_vector<4>(batch_dims); in ComputeWYRepresentation() local
|