Searched refs:with_rank (Results 1 – 23 of 23) sorted by relevance
65 return [op.inputs[0].get_shape().with_rank(rank)]106 a_shape = op.inputs[0].get_shape().with_rank(2)108 b_shape = op.inputs[1].get_shape().with_rank(2)190 input_shape = op.inputs[0].get_shape().with_rank(4)191 filter_shape = op.inputs[1].get_shape().with_rank(4)259 input_shape = op.inputs[0].get_shape().with_rank(4)260 filter_shape = op.inputs[1].get_shape().with_rank(4)319 input_shape = op.inputs[0].get_shape().with_rank(4)377 input_shape = op.inputs[0].get_shape().with_rank(4)445 input_shape = op.inputs[0].get_shape().with_rank(4)
130 indices_shape = indices.get_shape().with_rank(2)131 values_shape = values.get_shape().with_rank(1)132 dense_shape_shape = dense_shape.get_shape().with_rank(1)
954 def with_rank(self, rank): member in TensorShape
858 shape = tensor.get_shape().with_rank(1)
119 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value194 batch_size = x[0].get_shape().with_rank(2).dims[0].value195 cell_size4 = b.get_shape().with_rank(1).dims[0].value245 batch_size = x.get_shape().with_rank(2).dims[0].value248 input_size = x.get_shape().with_rank(2).dims[1].value251 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value516 inputs_shape = inputs.get_shape().with_rank(3)681 inputs_shape = inputs.get_shape().with_rank(3)
200 cell_size = h_prev.get_shape().with_rank(2)[1]
255 input_size = inputs.get_shape().with_rank(2).dims[1]486 input_size = input_feat.get_shape().with_rank(2).dims[-1].value893 input_size = input_feat.get_shape().with_rank(2).dims[-1].value1297 input_size = inp.get_shape().with_rank(2).dims[1].value1520 tensor_shape.TensorShape(inputs_shape).with_rank(2)[1])1694 input_size = inputs.get_shape().with_rank(2).dims[1]1805 input_size = inputs.get_shape().with_rank(2).dims[1]2670 input_size = inputs.get_shape().with_rank(2).dims[1]3032 input_size = inputs.get_shape().with_rank(2).dims[1]
70 name: "with_rank"
211 input_shape = input_shape.with_rank(2)
286 tensor_shape.dimension_value(inputs.shape.with_rank(2)[1]) > 0 and297 input_splits[0].shape.with_rank(2)[1])
2472 multiples_shape = op.inputs[1].get_shape().with_rank(1)2473 input_shape = op.inputs[0].get_shape().with_rank(multiples_shape[0])2477 multiples = tensor_util.constant_value_as_shape(op.inputs[1]).with_rank(2692 result.set_shape(result.get_shape().with_rank(4))2750 result.set_shape(result.get_shape().with_rank(4))
165 filter_shape = filter_shape.with_rank(input_shape.ndims)168 input_shape = input_shape.with_rank(filter_shape.ndims)511 rate_shape = dilation_rate.get_shape().with_rank(1)1025 input_shape.with_rank(num_spatial_dims + 2)1031 filter_shape.with_rank(num_spatial_dims + 2)1183 input.get_shape().with_rank(num_spatial_dims + 2)
115 static_shape = image.get_shape().with_rank(rank).as_list()138 image_shape = image.get_shape().with_rank(3)218 image_shape = image.get_shape().with_rank(3)
705 pointwise_filter_shape = pointwise_filter.get_shape().with_rank(4)
1013 input_size = inputs.get_shape().with_rank(2).dims[1].value
204 input_.set_shape(input_.get_shape().with_rank(2))
307 input_size = inputs.get_shape().with_rank(2).dims[1]
439 input_size = inputs.shape.with_rank(2)[1]
87 images.shape.with_rank(4)
683 input_size = inp.get_shape().with_rank(2)[1]
515 rank = t.dense_shape.shape.with_rank(1).dims[0]
172 new_value.set_shape(new_value.get_shape().with_rank(expected_rank))