Searched refs:with_rank (Results 1 – 15 of 15) sorted by relevance
144 indices_shape = indices.shape.with_rank(2)145 values_shape = values.shape.with_rank(1)146 dense_shape_shape = dense_shape.shape.with_rank(1)
237 self._indices_shape = tensor_shape.as_shape(indices_shape).with_rank(1)
1040 def with_rank(self, rank): member in TensorShape
922 shape = tensor.get_shape().with_rank(1)
70 name: "with_rank"
231 input_shape = input_shape.with_rank(2)
3696 multiples_shape = op.inputs[1].get_shape().with_rank(1)3697 input_shape = op.inputs[0].get_shape().with_rank(multiples_shape[0])3701 multiples = tensor_util.constant_value_as_shape(op.inputs[1]).with_rank(3966 result.set_shape(result.get_shape().with_rank(4))4031 result.set_shape(result.get_shape().with_rank(4))
117 static_shape = image.get_shape().with_rank(rank).as_list()140 image_shape = image.get_shape().with_rank(3)220 image_shape = image.get_shape().with_rank(3)286 image_shape = image.get_shape().with_rank(2)
198 filter_shape = filter_shape.with_rank(205 input_shape = input_shape.with_rank(1432 input.get_shape().with_rank(num_spatial_dims + 2)
1054 pointwise_filter_shape = pointwise_filter.get_shape().with_rank(4)
451 input_size = inputs.shape.with_rank(2).dims[1]
1059 uniform_row_length = uniform_row_length.with_rank(1)
514 rank = t.dense_shape.shape.with_rank(1).dims[0]
1047 input_size = inputs.get_shape().with_rank(2).dims[1].value