| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_factory_ops.py | 34 def constant(pylist, dtype=None, ragged_rank=None, inner_shape=None, name=None): argument
76 inner_shape)
80 def constant_value(pylist, dtype=None, ragged_rank=None, inner_shape=None): argument
131 ragged_rank, inner_shape)
135 inner_shape): argument
163 if inner_shape is not None and inner_shape:
166 (pylist, len(inner_shape)))
182 if inner_shape is not None and ragged_rank is not None:
183 expected_depth = ragged_rank + len(inner_shape) + 1
188 "and dim(inner_shape)=%d" % (pylist, ragged_rank, len(inner_shape)))
[all …]
|
| D | ragged_const_op_test.py | 61 inner_shape=(2,),
66 inner_shape=(2,),
82 inner_shape=(2,),
87 inner_shape=(2, 2),
102 dict(pylist=[], inner_shape=(0, 100, 20), expected_shape=(0, 100, 20)),
106 inner_shape=(100, 20),
111 inner_shape=(100, 20),
114 dict(pylist=[], inner_shape=(0,), expected_shape=(0,)),
115 dict(pylist=[[]], inner_shape=(1, 0), expected_shape=(1, 0)),
143 inner_shape=None, argument
[all …]
|
| D | ragged_constant_value_op_test.py | 60 inner_shape=(2,),
65 inner_shape=(2,),
81 inner_shape=(2,),
86 inner_shape=(2, 2),
101 dict(pylist=[], inner_shape=(0, 100, 20), expected_shape=(0, 100, 20)),
105 inner_shape=(100, 20),
110 inner_shape=(100, 20),
113 dict(pylist=[], inner_shape=(0,), expected_shape=(0,)),
114 dict(pylist=[[]], inner_shape=(1, 0), expected_shape=(1, 0)),
147 inner_shape=None, argument
[all …]
|
| D | ragged_gather_op_test.py | 108 inner_shape=(2,))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.ragged.pbtxt | 9 …argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'name\'], varargs=None, …
13 …argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\'], varargs=None, keywords=N…
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | reconstruction_ops.py | 70 def full_shape(inner_shape): argument
71 return array_ops.concat([outer_dimensions, inner_shape], 0)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.ragged.pbtxt | 5 …argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'name\'], varargs=None, …
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 915 Shape inner_shape = DropFirstDim(outer_array.GetShape()); in SliceOutInnerArray() local
916 std::vector<llvm::Value*> multidim_index(inner_shape.rank() + 1, in SliceOutInnerArray()
923 llvm_ir::ShapeToIrType(inner_shape, module)->getPointerTo(); in SliceOutInnerArray()
925 std::move(inner_shape)); in SliceOutInnerArray()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 446 inner_shape = params_shape[axis:][1:]
447 inner_dims = array_ops.size(inner_shape)
453 values_shape = array_ops.concat([outer_shape, indices_size, inner_shape], 0)
|