Searched refs:shape_prefix (Results 1 – 2 of 2) sorted by relevance
539 TensorShape shape_prefix = v.shape(); in Compute() local540 shape_prefix.RemoveDim(shape_prefix.dims() - 1); in Compute()549 common_prefix = shape_prefix; in Compute()552 OP_REQUIRES(ctx, shape_prefix == common_prefix, in Compute()
732 const TensorShape& shape_prefix = buf_tensor.shape(); in Compute() local734 TensorShape sizes_shape = shape_prefix; in Compute()791 TensorShape out_shape = shape_prefix; in Compute()