Searched defs:input2_slice_shape (Results 1 – 1 of 1) sorted by relevance
1893 Shape input2_slice_shape = inputs[2].slice_shape(); in GetForwardComputationCost() local1942 Shape input2_slice_shape = inputs[2].slice_shape(); // equal to output shape in GetBackwardComputationCost() local2015 Shape input2_slice_shape = inputs.at(2).slice_shape(); in GetForwardComputationCost() local