| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | StatefulStandardNormal.pbtxt | 9 type_attr: "shape_dtype"
31 name: "shape_dtype"
53 type_attr: "shape_dtype"
67 name: "shape_dtype"
83 type_attr: "shape_dtype"
97 name: "shape_dtype"
|
| D | NonDeterministicInts.pbtxt | 5 type_attr: "shape_dtype"
19 name: "shape_dtype"
|
| D | StatefulStandardNormalV2.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulUniformFullInt.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulUniform.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulTruncatedNormal.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulUniformInt.pbtxt | 13 type_attr: "shape_dtype"
35 name: "shape_dtype"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | StatefulStandardNormal.pbtxt | 9 type_attr: "shape_dtype"
31 name: "shape_dtype"
53 type_attr: "shape_dtype"
67 name: "shape_dtype"
83 type_attr: "shape_dtype"
97 name: "shape_dtype"
|
| D | NonDeterministicInts.pbtxt | 5 type_attr: "shape_dtype"
19 name: "shape_dtype"
|
| D | StatefulUniform.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulTruncatedNormal.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulUniformFullInt.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulStandardNormalV2.pbtxt | 13 type_attr: "shape_dtype"
27 name: "shape_dtype"
|
| D | StatefulUniformInt.pbtxt | 13 type_attr: "shape_dtype"
35 name: "shape_dtype"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 146 IntegerType shape_dtype = rewriter->getIntegerType(32); in CreateSliceOpForTensorList() local
147 RankedTensorType position_type = RankedTensorType::get({-1}, shape_dtype); in CreateSliceOpForTensorList()
151 RankedTensorType vector_type = RankedTensorType::get({1}, shape_dtype); in CreateSliceOpForTensorList()
269 IntegerType shape_dtype = rewriter.getIntegerType(32); in matchAndRewrite() local
271 loc, RankedTensorType::get({}, shape_dtype), item); in matchAndRewrite()
281 loc, RankedTensorType::get({1}, shape_dtype), item_rank, scalar_zero); in matchAndRewrite()
342 Type shape_dtype = getElementTypeOrSelf(element_shape.getType()); in matchAndRewrite() local
355 op.getLoc(), RankedTensorType::get({-1}, shape_dtype), in matchAndRewrite()
460 Type shape_type = RankedTensorType::get({result_rank}, shape_dtype); in matchAndRewrite()
489 Type shape_dtype = getElementTypeOrSelf(op.element_shape().getType()); in GetNumElements() local
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 108 def wrap(op, dtype, shape, shape_dtype, kwds, seed): argument
114 shape_ = (constant_op.constant(shape, dtype=shape_dtype)
115 if shape_dtype is not None else shape)
125 for shape_dtype in shape_dtypes:
129 (name, _name(dtype), shape, _name(shape_dtype))).replace(
132 shape_dtype, kwds),
133 functools.partial(wrap, stateful_op, dtype, shape, shape_dtype,
139 def wrap(op, minval, maxval, shape, shape_dtype, dtype, seed): argument
140 shape_ = (constant_op.constant(shape, dtype=shape_dtype)
141 if shape_dtype is not None else shape)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 221 shape_dtype = _find_shape_dtype(fields, nrows, row_partitions)
223 nrows = ops.convert_to_tensor(nrows, shape_dtype)
240 shape_dtype, validate)
246 nrows = constant_op.constant(static_nrows.value, shape_dtype)
252 shape_dtype, validate)
259 np.array(shape.as_list(), dtype=shape_dtype.as_numpy_dtype),
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1517 TF_DerivedOperandTypeAttr shape_dtype = TF_DerivedOperandTypeAttr<2>;
1540 TF_DerivedOperandTypeAttr shape_dtype = TF_DerivedOperandTypeAttr<2>;
1562 TF_DerivedOperandTypeAttr shape_dtype = TF_DerivedOperandTypeAttr<2>;
1583 TF_DerivedOperandTypeAttr shape_dtype = TF_DerivedOperandTypeAttr<2>;
1615 TF_DerivedOperandTypeAttr shape_dtype = TF_DerivedOperandTypeAttr<2>;
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 26360 type_attr: "shape_dtype"
26374 name: "shape_dtype"
51334 type_attr: "shape_dtype"
51348 name: "shape_dtype"
51372 type_attr: "shape_dtype"
51386 name: "shape_dtype"
51406 type_attr: "shape_dtype"
51420 name: "shape_dtype"
51440 type_attr: "shape_dtype"
51454 name: "shape_dtype"
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 79258 type_attr: "shape_dtype"
79280 name: "shape_dtype"
79302 type_attr: "shape_dtype"
79316 name: "shape_dtype"
79336 type_attr: "shape_dtype"
79350 name: "shape_dtype"
79370 type_attr: "shape_dtype"
79384 name: "shape_dtype"
79404 type_attr: "shape_dtype"
79426 name: "shape_dtype"
|