Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/contrib/crf/python/ops/
Dcrf.py112 tensor_shape.dimension_value(
164 tensor_shape.dimension_value(
223 tensor_shape.dimension_value(
250 num_tags = tensor_shape.dimension_value(inputs.shape[2])
352 self._num_tags = tensor_shape.dimension_value(transition_params.shape[0])
438 self._num_tags = tensor_shape.dimension_value(transition_params.shape[0])
550 num_tags = tensor_shape.dimension_value(potentials.shape[2])
593 pred=math_ops.equal(tensor_shape.dimension_value(potentials.shape[1]) or
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dcore_layers.py128 if tensor_shape.dimension_value(input_shape[channel_axis]) is None:
131 input_dim = tensor_shape.dimension_value(input_shape[channel_axis])
401 if tensor_shape.dimension_value(input_shape[-1]) is None:
405 min_ndim=2, axes={-1: tensor_shape.dimension_value(input_shape[-1])})
409 shape=[tensor_shape.dimension_value(input_shape[-1]), self.units],
417 shape=[tensor_shape.dimension_value(input_shape[-1]), self.units],
474 if tensor_shape.dimension_value(input_shape[-1]) is None:
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
Dtest_utils.py39 state_dimension = tensor_shape.dimension_value(transition_matrix.shape[0])
67 state_dimension = tensor_shape.dimension_value(transition_matrix.shape[0])
68 state_noise_dimension = tensor_shape.dimension_value(noise_transform.shape[1])
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dfill_triangular.py110 tensor_shape.dimension_value(input_shape[-1]))
119 tensor_shape.dimension_value(output_shape[-2]),
120 tensor_shape.dimension_value(output_shape[-1]))
Dmasked_autoregressive.py241 event_size = tensor_shape.dimension_value(
265 static_event_size = tensor_shape.dimension_value(
411 input_depth = tensor_shape.dimension_value(
527 input_depth = tensor_shape.dimension_value(
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py219 n = (tensor_shape.dimension_value(static_shape[0])
271 n_static = tensor_shape.Dimension(tensor_shape.dimension_value(
274 n_static.merge_with(tensor_shape.Dimension(tensor_shape.dimension_value(
Dfunctional_ops.py123 n = (tensor_shape.dimension_value(elems_flat[0].shape[0])
232 n = (tensor_shape.dimension_value(elems_flat[0].shape[0])
400 n = tensor_shape.dimension_value(elems_flat[0].shape[0])
479 n_static = tensor_shape.Dimension(tensor_shape.dimension_value(
482 n_static.merge_with(tensor_shape.Dimension(tensor_shape.dimension_value(
Dembedding_ops.py158 dim_0_size = tensor_shape.Dimension(tensor_shape.dimension_value(
161 dim_0_size += tensor_shape.Dimension(tensor_shape.dimension_value(
168 param_p_dim = tensor_shape.dimension_value(params[p].get_shape()[0])
763 original_rank_dim = tensor_shape.dimension_value(
Drnn.py1104 batch_size = tensor_shape.dimension_value(static_batch_size)
1322 if tensor_shape.dimension_value(size) is None:
1329 if tensor_shape.dimension_value(fixed_batch_size):
1330 batch_size = tensor_shape.dimension_value(fixed_batch_size)
1357 shape = _concat(tensor_shape.dimension_value(fixed_batch_size),
/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.py428 m = tensor_shape.dimension_value(maindiag.shape[-1])
431 n = tensor_shape.dimension_value(t.shape[-1])
449 m1 = tensor_shape.dimension_value(diagonals.shape[-1])
450 m2 = tensor_shape.dimension_value(diagonals.shape[-2])
Dlinear_operator.py406 dim_value = tensor_shape.dimension_value(self.domain_dimension)
445 dim_value = tensor_shape.dimension_value(self.range_dimension)
929 dim_value = tensor_shape.dimension_value(self.domain_dimension)
/external/google-fruit/extras/benchmark/
Dformat_bench_results.py26 params = {dimension_name: make_immutable(dimension_value)
27 for dimension_name, dimension_value in bench_result['benchmark'].items()}
319 …enchmark_params = {dimension_name: make_immutable(dimension_value) for dimension_name, dimension_v…
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dattention_wrapper.py154 tensor_shape.dimension_value(self._keys.shape[0]) or
156 self._alignments_size = (tensor_shape.dimension_value(self._keys.shape[1])
460 tensor_shape.dimension_value(self.keys.shape[0]) or
462 self._alignments_size = (tensor_shape.dimension_value(self.keys.shape[1])
1220 batch_size = tensor_shape.dimension_value(
2029 if tensor_shape.dimension_value(logits.get_shape()[-1]) is not None:
2030 depth = tensor_shape.dimension_value(logits.get_shape()[-1])
2233 tensor_shape.dimension_value(layer.compute_output_shape(
2235 cell.output_size + tensor_shape.dimension_value(
2242 tensor_shape.dimension_value(attention_mechanism.values.shape[-1])
[all …]
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dindependent.py169 dim0 = tensor_shape.dimension_value(
187 dim0 = tensor_shape.dimension_value(
246 dim0 = tensor_shape.dimension_value(
Dmixture_same_family.py151 s_dim0 = tensor_shape.dimension_value(s.shape[0])
191 km = tensor_shape.dimension_value(
193 kc = tensor_shape.dimension_value(
Dvector_diffeomixture.py186 num_components = tensor_shape.dimension_value(bs[-1])
794 deg = tensor_shape.dimension_value(p.shape.with_rank_at_least(1)[-1])
834 if tensor_shape.dimension_value(param.shape[-1]) != 1:
838 tensor_shape.dimension_value(
910 deg = tensor_shape.dimension_value(grid.shape.with_rank_at_least(1)[-1])
944 deg = tensor_shape.dimension_value(grid.shape.with_rank_at_least(1)[-1])
/external/tensorflow/tensorflow/python/framework/
Dtensor_shape_test.py259 assert tensor_shape.dimension_value(d1) == d2
275 assert tensor_shape.dimension_value(d1) == d2
305 tensor_shape.dimension_value(unknown[2]))
380 self.assertEqual(tensor_shape.dimension_value(partial_proto_shape[0]), None)
381 self.assertEqual(tensor_shape.dimension_value(partial_proto_shape[1]), 37)
382 self.assertEqual(tensor_shape.dimension_value(partial_proto_shape[2]), 42)
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dgru_ops.py180 input_size = tensor_shape.dimension_value(input_shape[1])
225 input_size = tensor_shape.dimension_value(input_shape[1])
Drnn_cell.py642 batch_size = tensor_shape.dimension_value(
1065 batch_size = tensor_shape.dimension_value(
1519 input_size = tensor_shape.dimension_value(
2382 self._batch_size = tensor_shape.dimension_value(
2387 input_size = tensor_shape.dimension_value(inputs.shape[1])
2608 if tensor_shape.dimension_value(shape[1]) is None:
2612 total_arg_size += tensor_shape.dimension_value(shape[1])
2781 if tensor_shape.dimension_value(inputs_shape[1]) is None:
2785 input_depth = tensor_shape.dimension_value(inputs_shape[1])
2958 if tensor_shape.dimension_value(shape[1]) is None:
[all …]
/external/tensorflow/tensorflow/contrib/image/python/ops/
Dinterpolate_spline.py104 if tensor_shape.dimension_value(d) is None:
109 if tensor_shape.dimension_value(k) is None:
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
Dsdca_ops.py443 if tensor_shape.dimension_value(dim_0_size):
445 tensor_shape.dimension_value(dim_0_size),
450 if tensor_shape.dimension_value(w[p].shape[0]) is not None:
451 dim_0_sizes.append(tensor_shape.dimension_value(w[p].shape[0]))
/external/tensorflow/tensorflow/python/ops/ragged/
Dsegment_id_ops.py58 if tensor_shape.dimension_value(splits.shape[0]) == 0:
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py354 if tensor_shape.dimension_value(x_shape_static[-1]) is not None:
834 if tensor_shape.dimension_value(
947 if tensor_shape.dimension_value(
1202 s = tensor_shape.dimension_value(
1253 return tensor_shape.dimension_value(x.shape.with_rank_at_least(1)[-1])
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.compat.pbtxt40 name: "dimension_value"

123