Searched refs:chip_shape (Results 1 – 3 of 3) sorted by relevance
33 TensorShape chip_shape = parent.shape(); in ValidateInput() local34 chip_shape.RemoveDim(0); in ValidateInput()39 ", [parent slice]: ", chip_shape.DebugString()); in ValidateInput()318 TensorShape chip_shape = parent->shape(); in ValidateElementToLargerSlice() local319 chip_shape.RemoveDim(0); in ValidateElementToLargerSlice()324 ", [parent slice]: ", chip_shape.DebugString()); in ValidateElementToLargerSlice()
37 TensorShape chip_shape = parent.shape(); in HandleSliceToElement() local38 chip_shape.RemoveDim(0); in HandleSliceToElement()43 ", [parent slice]: ", chip_shape.DebugString()); in HandleSliceToElement()
300 TensorShape chip_shape = parent->shape(); in ValidateElementToLargerSlice() local301 chip_shape.RemoveDim(0); in ValidateElementToLargerSlice()306 ", [parent slice]: ", chip_shape.DebugString()); in ValidateElementToLargerSlice()