| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_const_op_test.py | 61 inner_shape=(2,),
66 inner_shape=(2,),
81 inner_shape=(2,),
87 inner_shape=(2,),
103 inner_shape=(2,),
108 inner_shape=(2, 2),
132 dict(pylist=[], inner_shape=(0, 100, 20), expected_shape=(0, 100, 20)),
136 inner_shape=(100, 20),
141 inner_shape=(100, 20),
144 dict(pylist=[], inner_shape=(0,), expected_shape=(0,)),
[all …]
|
| D | ragged_factory_ops.py | 39 def constant(pylist, dtype=None, ragged_rank=None, inner_shape=None, argument
87 ragged_rank, inner_shape)
92 def constant_value(pylist, dtype=None, ragged_rank=None, inner_shape=None, argument
147 ragged_rank, inner_shape)
151 inner_shape): argument
179 if inner_shape is not None and inner_shape:
182 (pylist, len(inner_shape)))
198 if inner_shape is not None and ragged_rank is not None:
199 expected_depth = ragged_rank + len(inner_shape) + 1
204 "and dim(inner_shape)=%d" % (pylist, ragged_rank, len(inner_shape)))
[all …]
|
| D | ragged_constant_value_op_test.py | 59 inner_shape=(2,),
64 inner_shape=(2,),
79 inner_shape=(2,),
85 inner_shape=(2,),
101 inner_shape=(2,),
106 inner_shape=(2, 2),
130 dict(pylist=[], inner_shape=(0, 100, 20), expected_shape=(0, 100, 20)),
134 inner_shape=(100, 20),
139 inner_shape=(100, 20),
142 dict(pylist=[], inner_shape=(0,), expected_shape=(0,)),
[all …]
|
| D | ragged_to_tensor_op_test.py | 328 inner_shape=None, argument
333 rt_input, ragged_rank=ragged_rank, inner_shape=inner_shape)
|
| D | ragged_tensor_test.py | 1612 inner_shape=(2,), dtype=np.int32),
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 495 inner_shape = shape[-inner_rank:]
496 if not inner_shape.is_fully_defined():
497 inner_shape = array_ops.shape(inp)[-inner_rank:]
503 if isinstance(inner_shape, tensor_shape.TensorShape):
504 inp_reshaped = array_ops.reshape(inp, [-1] + inner_shape.as_list())
507 inp, array_ops.concat(([-1], inner_shape), axis=-1))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.ragged.pbtxt | 13 …argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'name\', \'row_splits_dt…
17 …argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'row_splits_dtype\'], va…
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | reconstruction_ops.py | 81 def full_shape(inner_shape): argument
82 return array_ops.concat([outer_dimensions, inner_shape], 0)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.ragged.pbtxt | 9 …argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'name\', \'row_splits_dt…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 616 inner_shape = values_shape[batch_dims:][1:]
618 flat_values_shape = array_ops.concat([[-1], inner_shape], 0)
675 inner_shape = params_shape[axis:][1:]
676 values_shape = array_ops.concat([outer_shape, [-1], inner_shape], 0)
|
| D | nn_ops.py | 299 inner_shape = shape[-inner_rank:]
300 if not inner_shape.is_fully_defined():
301 inner_shape = array_ops.shape(inp)[-inner_rank:]
307 if isinstance(inner_shape, tensor_shape.TensorShape):
308 inp_reshaped = array_ops.reshape(inp, [-1] + inner_shape.as_list())
311 inp, array_ops.concat(([-1], inner_shape), axis=-1))
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1111 Shape inner_shape = DropFirstDim(outer_array.GetShape()); in SliceOutInnerArray() local
1112 std::vector<llvm::Value*> multidim_index(inner_shape.rank() + 1, in SliceOutInnerArray()
1119 llvm_ir::ShapeToIrType(inner_shape, module)->getPointerTo(); in SliceOutInnerArray()
1121 std::move(inner_shape)); in SliceOutInnerArray()
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 919 inner_shape=typespec._shape[typespec._ragged_rank + 1:])
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | text_vectorization_test.py | 429 output = vectorization(ragged_factory_ops.constant(data, inner_shape=(1,)))
|