Searched refs:second_shape (Results 1 – 3 of 3) sorted by relevance
195 auto second_shape = inputs[1]->tensor.shape; in GPUOperationFromNode() local201 BHWC weights_shape(second_shape.c, 1, 1, second_shape.w); in GPUOperationFromNode()
744 TensorShape second_shape; in AllocateOutputSetMklShape() local745 second_shape.AddDim(mkl_shape.GetSerializeBufferSize()); in AllocateOutputSetMklShape()748 second_shape, &second_tensor)); in AllocateOutputSetMklShape()766 TensorShape second_shape; variable767 second_shape.AddDim(mkl_shape.GetSerializeBufferSize());770 second_shape, &second_tensor));
1275 auto second_shape = second_type.getShape(); in Verify() local1277 if (first_shape[d] != second_shape[d] && d != op.dimension()) { in Verify()1282 llvm::make_range(second_shape.begin(), second_shape.end()))); in Verify()