Searched defs:second_shape (Results 1 – 3 of 3) sorted by relevance
491 const BHWC second_shape(2, 2, 2, 2); // dummy non-broadcasted shape in CreateGpuOperation() local497 const BHWC& second_shape) { in CreateGpuOperation()
754 TensorShape second_shape; in AllocateOutputSetMklShape() local776 TensorShape second_shape; variable
325 auto second_shape = inputs[1]->tensor.shape; in GPUOperationFromNodePart0() local