Searched refs:compute_shape (Results 1 – 2 of 2) sorted by relevance
901 TensorShape compute_shape = ShapeFromFormat( in operator ()() local906 ctx->allocate_temp(DataTypeToEnum<T>::value, compute_shape, in operator ()()913 CHECK(transformed_out_backprop.CopyFrom(out_backprop, compute_shape)); in operator ()()922 TensorShape compute_shape = ShapeFromFormat( in operator ()() local927 if (compute_shape.dim_size(1) > 1) { in operator ()()929 ctx->allocate_temp(DataTypeToEnum<T>::value, compute_shape, in operator ()()937 CHECK(transformed_input.CopyFrom(compatible_input, compute_shape)); in operator ()()
316 TensorShape compute_shape = ShapeFromFormat( in operator ()() local321 ctx->allocate_temp(DataTypeToEnum<T>::value, compute_shape, in operator ()()328 CHECK(transformed_out_backprop.CopyFrom(out_backprop, compute_shape)); in operator ()()