Searched defs:biasShape (Results 1 – 12 of 12) sorted by relevance
43 std::vector<int32_t> biasShape { 1 }; in Conv3DWithBiasesSimpleWithPaddingFp32Test() local84 std::vector<int32_t> biasShape { 1 }; in Conv3DWithBiasesStridesFp32Test() local143 std::vector<int32_t> biasShape { 2 }; in Conv3DWithBiasesDilationFp32Test() local194 std::vector<int32_t> biasShape { 1 }; in Conv3DFp32SmallTest() local
27 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesFp32Test() local81 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesInt8Test() local124 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesReluUint8Test() local182 std::vector<int32_t> biasShape { 1 }; in Conv2DWithBiasesRelu6Uint8Test() local231 std::vector<int32_t> biasShape { 4 }; in Conv2DPerChannelInt8Test() local
27 std::vector<int32_t> biasShape { 4 }; in DepthwiseConv2dValidReluFp32Test() local89 std::vector<int32_t> biasShape { 1 } ; in DepthwiseConv2dSameUint8Test() local136 std::vector<int32_t> biasShape {16} ; in DepthwiseConv2dSameInt8PerChannelTest() local
212 const std::vector<int32_t>& biasShape = {}, in ConvolutionTest()448 const std::vector<int32_t>& biasShape = {}, in Convolution3dTest()
32 const TensorShape biasShape{outputSize}; in FullyConnected() local
60 const TensorShape biasShape{outChannels}; in Convolve3d() local
117 const TensorShape biasShape{outputChannels}; in Convolve() local
200 unsigned int biasShape[] = { outputChannels }; in FullyConnectedLargeTestCommon() local301 unsigned int biasShape[] = { outputChannels }; in FullyConnectedFloat32Test() local
1388 unsigned int biasShape[] = {outputChannels}; in CompareConvolution2dTestImpl() local3065 std::vector<unsigned int> biasShape{ outputChannels }; in CompareDepthwiseConvolution2dTestImpl() local
151 unsigned int biasShape[] = { 1, outputChannels, outputHeight, outputWidth }; variable655 const TensorShape biasShape { cOutput }; variable700 const TensorShape biasShape { cOutput }; variable746 const TensorShape biasShape { cOutput }; variable
242 unsigned int biasShape[] = { outputChannels }; in FullyConnectedWithDynamicOrConstantInputsEndToEnd() local358 unsigned int biasShape[] = { outputChannels }; in FullyConnectedErrorChecking() local
613 std::vector<ExprHandle> biasShape; in computeValue() local