Searched defs:input1_slice_shape (Results 1 – 2 of 2) sorted by relevance
113 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local135 Shape input1_slice_shape = inputs[1].slice_shape(); in GetForwardComputationCost() local159 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardComputationCost() local232 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local293 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local359 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local596 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local614 Shape input1_slice_shape = inputs[1].slice_shape(); in GetForwardComputationCost() local634 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardComputationCost() local722 Shape input1_slice_shape = inputs[1].slice_shape(); in GetForwardComputationCost() local[all …]
63 Shape input0_slice_shape{20, 50}, input1_slice_shape{50, 25}, output0_slice_shape{20, 25}; in TEST_F() local