Home
last modified time | relevance | path

Searched refs:merge_with (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dtensor_shape_test.py55 dim.merge_with(tensor_shape.Dimension(12)))
56 self.assertEqual(tensor_shape.Dimension(12), dim.merge_with(12))
68 dim.merge_with(tensor_shape.Dimension(13))
82 dim.merge_with(tensor_shape.Dimension(None)).value)
108 tensor_shape.Dimension(12), known.merge_with(unknown))
110 tensor_shape.Dimension(12), unknown.merge_with(known))
181 tensor_shape.Dimension(42).merge_with([])
184 tensor_shape.Dimension(42).merge_with(-1)
299 tensor_shape.TensorShape([3, 4, 7]).merge_with(
302 tensor_shape.TensorShape([3, 4, 7]).merge_with(
[all …]
Dtensor_shape.py288 def merge_with(self, other): member in Dimension
929 def merge_with(self, other): member in TensorShape
974 dim.merge_with(other_dim)
1056 return self.merge_with(unknown_shape(rank=rank))
Dcommon_shapes.py67 return_dims.append(dim_x.merge_with(dim_y))
Dindexed_slices.py170 indices_shape = self._indices.shape.merge_with(self._values.shape[:1])
175 dense_shape = dense_shape.merge_with(
Dtensor_spec.py207 if batched and self._shape.merge_with(value.shape).ndims == 0:
Dsparse_tensor.py367 if self._shape.merge_with(dense_shape).ndims == 0:
Dtensor_util.py1025 ret = ret.merge_with(
/external/tensorflow/tensorflow/core/kernels/
Disotonic_regression_op.cc50 void merge_with(const Segment& other) { in merge_with() function in __anon733d8bd10111::Segment
79 void merge_with(const L2PavaSegment& other) { in merge_with() function in __anon733d8bd10111::L2PavaSegment
80 Segment::merge_with(other); in merge_with()
126 (pools.rbegin() + 1)->merge_with(*pools.rbegin()); in solve_pava()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-dimension.pbtxt22 name: "merge_with"
Dtensorflow.-tensor-shape.pbtxt58 name: "merge_with"
/external/tensorflow/tensorflow/python/training/
Dinput.py181 element_shape = input_tensor.shape[1:].merge_with(element_shape)
424 def merge_with(self, other): member in _SparseMetaData
429 self.rank.merge_with(other.rank)
593 info.merge_with(candidate)
677 merged_shape.merge_with(s)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-tensor-shape.pbtxt58 name: "merge_with"
/external/tensorflow/tensorflow/python/ops/distributions/
Ddistribution.py1290 x.set_shape(x.get_shape().merge_with(shape))
1296 x.set_shape(x.get_shape().merge_with(shape))
1308 x.set_shape(x.get_shape().merge_with(shape))
/external/tensorflow/tensorflow/python/ops/
Dlinalg_grad.py837 m = a_shape.dims[-2].merge_with(grad_u_shape[-2])
838 n = a_shape.dims[-1].merge_with(grad_v_shape[-2])
839 batch_shape = a_shape[:-2].merge_with(grad_u_shape[:-2]).merge_with(
Dtensor_array_ops.py200 self._element_shape[0] = self.element_shape.merge_with(shape)
508 self._element_shape[0] = self.element_shape.merge_with(shape)
827 self._element_shape = self._element_shape.merge_with(value.shape)
Dembedding_ops.py223 element_shape_s = element_shape_s.merge_with(p.get_shape()[1:])
Darray_grad.py431 input_shape = op.inputs[0].get_shape().merge_with(grad.get_shape())
433 batch_shape = input_shape[:-2].merge_with(diag_shape[:-1])
Dgradients_util.py843 shape = shape.merge_with(i.get_shape())
Dlookup_ops.py175 self._default_value.get_shape().merge_with(tensor_shape.TensorShape([]))
Ddata_flow_ops.py392 batch_dim = batch_dim.merge_with(val_batch_dim)
/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition.py1099 nrows = self._nrows.merge_with(other.nrows)
1100 nvals = self._nvals.merge_with(other.nvals)
1101 ncols = self._uniform_row_length.merge_with(other.uniform_row_length)
/external/tensorflow/tensorflow/python/profiler/
Dtfprof_logger.py60 op.outputs[i].set_shape(op.outputs[i].get_shape().merge_with(
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor.py233 shape = shape.merge_with(value.shape[:rank])
1233 return nrows, static_nrows.merge_with(static_value_nrows)
/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce.py48 shape = shape.merge_with(tensor.shape)
/external/tensorflow/tensorflow/python/keras/engine/
Dfunctional.py637 tensor.set_shape(tensor.shape.merge_with(ref_input.shape))

12