Searched defs:input1_slice_shape (Results 1 – 3 of 3) sorted by relevance
113 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local134 Shape input1_slice_shape = inputs[1].slice_shape(); in GetForwardComputationCost() local158 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardComputationCost() local214 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local280 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local517 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local535 Shape input1_slice_shape = inputs[1].slice_shape(); in GetForwardComputationCost() local555 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardComputationCost() local643 Shape input1_slice_shape = inputs[1].slice_shape(); in GetForwardComputationCost() local691 Shape input1_slice_shape = input1.slice_shape(); in GetBackwardCommCost() local[all …]
63 Shape input0_slice_shape{20, 50}, input1_slice_shape{50, 25}, output0_slice_shape{20, 25}; in TEST_F() local
595 Shape input1_slice_shape = input1_info.slice_shape(); in TEST_F() local