Searched refs:x_rank (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 108 absl::string_view einsum_config, int64 x_rank, int64 y_rank);
|
| D | matrix.cc | 378 const int64 x_rank = x_config.size(); in Einsum() local
413 for (int64 i = 0; i < x_rank; ++i) { in Einsum()
559 absl::string_view einsum_config, int64 x_rank, int64 y_rank) { in ParseEinsumString() argument
621 /*is_input_config=*/true, x_rank, in ParseEinsumString()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 1910 x_rank = None
1912 x_rank = x.shape.rank
1915 x_rank = x.dense_shape.shape.dims[0].value # sparse.dense_shape is 1-D.
1917 if x_rank:
1918 return constant_op.constant(np.arange(x_rank, dtype=np.int32))
|
| D | rnn.py | 60 x_rank = array_ops.rank(x)
62 x, array_ops.concat(([1, 0], math_ops.range(2, x_rank)), axis=0))
|