| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 46 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
73 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
89 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
105 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
116 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
151 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
171 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
194 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
210 const float* bias_data, const Dims<4>& bias_dims, in Conv()
236 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 141 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
173 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
189 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
205 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
284 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
324 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
344 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
367 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
608 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
621 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 119 memory::dims bias_dims = memory::dims({channel}); in Compute() local
|
| D | mkl_conv_ops.h | 304 memory::dims* bias_dims) { in GetBiasSizeInMklOrder()
|
| D | mkl_qmatmul_op.cc | 226 memory::dims bias_dims = {static_cast<int>(bias_tensor.dim_size(0))}; in Compute() local
|
| D | mkl_matmul_ops_common.h | 62 memory::dims bias_dims; member
|
| D | mkl_conv_ops.cc | 45 memory::dims bias_dims; member
787 memory::dims bias_dims = {}; in Compute() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | Matmul.cpp | 130 dnnl::memory::dims m1_dims, m2_dims, dst_dims, bias_dims; in matmul() local
|
| /external/XNNPACK/test/ |
| D | fully-connected.cc | 99 std::vector<size_t> bias_dims; member in FullyConnectedTestBase
|
| D | convolution-2d.cc | 100 std::array<size_t, 1> bias_dims; member in xnnpack::ConvolutionTestBase
|
| D | depthwise-convolution-2d.cc | 104 std::array<size_t, 1> bias_dims; member in xnnpack::DepthwiseConvolutionTestBase
|
| D | deconvolution-2d.cc | 104 std::array<size_t, 1> bias_dims; member in DeconvolutionTestBase
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3706 std::vector<int64_t> bias_dims = bias_descriptor.vectorized_dims( in GetCudnnFusedOperationGraph() local
|