Home
last modified time | relevance | path

Searched refs:dimension_value (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition.py1069 nrows = tensor_shape.dimension_value(nrows[0])
1070 nvals = tensor_shape.dimension_value(nvals[0])
1071 ncols = tensor_shape.dimension_value(uniform_row_length[0])
1111 nrows = tensor_shape.dimension_value(nrows[0])
1112 nvals = tensor_shape.dimension_value(nvals[0])
1117 return tensor_shape.dimension_value(self._nrows[0])
1121 return tensor_shape.dimension_value(self._nvals[0])
1125 return tensor_shape.dimension_value(self._uniform_row_length[0])
1158 nrows = tensor_shape.dimension_value(nrows[0])
1159 nvals = tensor_shape.dimension_value(nvals[0])
[all …]
Dsegment_id_ops.py64 if tensor_shape.dimension_value(splits.shape[0]) == 0:
Dragged_tensor_shape.py221 inner_ndims = tensor_shape.dimension_value(self._inner_dim_sizes.shape[0])
253 return tensor_shape.dimension_value(self._inner_dim_sizes.shape[0])
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_addition.py199 (op.name, tensor_shape.dimension_value(op.domain_dimension))
201 if tensor_shape.dimension_value(op.domain_dimension) is not None]
207 (op.name, tensor_shape.dimension_value(op.range_dimension))
209 if tensor_shape.dimension_value(op.range_dimension) is not None]
Dlinalg_impl.py563 m = tensor_shape.dimension_value(maindiag.shape[-1])
566 n = tensor_shape.dimension_value(t.shape[-1])
585 m1 = tensor_shape.dimension_value(diagonals.shape[-1])
586 m2 = tensor_shape.dimension_value(diagonals.shape[-2])
706 m1 = tensor_shape.dimension_value(diagonals.shape[-1])
707 m2 = tensor_shape.dimension_value(diagonals.shape[-2])
869 dim_val = tensor_shape.dimension_value(a.shape[dim])
Dlinear_operator_low_rank_update.py302 if tensor_shape.dimension_value(self.u.shape[-1]) is not None:
303 r = tensor_shape.dimension_value(self.u.shape[-1])
Dlinear_operator.py461 dim_value = tensor_shape.dimension_value(self.domain_dimension)
505 dim_value = tensor_shape.dimension_value(self.range_dimension)
1010 dim_value = tensor_shape.dimension_value(self.domain_dimension)
Dlinear_operator_identity.py79 domain_dim = tensor_shape.dimension_value(self.domain_dimension)
80 range_dim = tensor_shape.dimension_value(self.range_dimension)
/external/google-fruit/extras/benchmark/
Dformat_bench_results.py37 params = {dimension_name: make_immutable(dimension_value)
38 for dimension_name, dimension_value in bench_result['benchmark'].items()}
353 …enchmark_params = {dimension_name: make_immutable(dimension_value) for dimension_name, dimension_v…
390 params = {dimension_name: make_immutable(dimension_value)
391 for dimension_name, dimension_value in bench_result['benchmark'].items()}
/external/tensorflow/tensorflow/python/framework/
Dtensor_shape_test.py277 assert tensor_shape.dimension_value(d1) == d2
295 assert tensor_shape.dimension_value(d1) == d2
325 tensor_shape.dimension_value(unknown[2]))
424 self.assertEqual(tensor_shape.dimension_value(partial_proto_shape[0]), None)
425 self.assertEqual(tensor_shape.dimension_value(partial_proto_shape[1]), 37)
426 self.assertEqual(tensor_shape.dimension_value(partial_proto_shape[2]), 42)
Dsparse_tensor.py281 rank = tensor_shape.dimension_value(shape[0])
/external/tensorflow/tensorflow/python/ops/signal/
Ddct_ops.py105 tensor_shape.dimension_value(input.shape[-1]) or
117 axis_dim = (tensor_shape.dimension_value(input.shape[-1])
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py133 tensor_shape.dimension_value(elems_flat[0].shape[0]) or
328 tensor_shape.dimension_value(elems_flat[0].shape[0]) or
587 n = tensor_shape.dimension_value(elems_flat[0].shape[0])
675 tensor_shape.dimension_value(
680 tensor_shape.dimension_value(
Dembedding_ops.py163 tensor_shape.dimension_value(params[0].get_shape()[0]))
166 tensor_shape.dimension_value(params[p].get_shape()[0]))
172 param_p_dim = tensor_shape.dimension_value(params[p].get_shape()[0])
881 original_rank_dim = tensor_shape.dimension_value(
Drnn.py1118 batch_size = tensor_shape.dimension_value(static_batch_size)
1344 if tensor_shape.dimension_value(size) is None:
1351 if tensor_shape.dimension_value(fixed_batch_size):
1352 batch_size = tensor_shape.dimension_value(fixed_batch_size)
1380 tensor_shape.dimension_value(fixed_batch_size),
Dmap_fn.py437 tensor_shape.dimension_value(
442 tensor_shape.dimension_value(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.compat.pbtxt40 name: "dimension_value"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.compat.pbtxt40 name: "dimension_value"
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py365 if tensor_shape.dimension_value(x_shape_static[-1]) is not None:
843 if tensor_shape.dimension_value(
953 if tensor_shape.dimension_value(
1209 s = tensor_shape.dimension_value(
1261 return tensor_shape.dimension_value(x.shape.with_rank_at_least(1)[-1])
Dcategorical.py217 if tensor_shape.dimension_value(logits_shape_static[-1]) is not None:
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py177 dynamic_size=tensor_shape.dimension_value(flat_shape[0]),
178 infer_shape=tensor_shape.dimension_value(flat_shape[1])))
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dload.py1099 or tensor_shape.dimension_value(dim_x) is None
1100 or tensor_shape.dimension_value(dim_y) is None):
1103 dims.append(tensor_shape.dimension_value(dim_x))
/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce_test.py121 tlen = tensor_shape.dimension_value(input_tensors[0].shape[0])
/external/tensorflow/tensorflow/python/keras/layers/
Dcore.py681 batch_dim = tensor_shape.dimension_value(input_shape[0])
1181 last_dim = tensor_shape.dimension_value(input_shape[-1])
1218 if tensor_shape.dimension_value(input_shape[-1]) is None:
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py233 none_filler if tensor_shape.dimension_value(d) is None else d

12