| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_ops_test.py | 42 index_dtype, argument
51 index = array_ops.placeholder(index_dtype)
96 def testReduceSumF32(self, index_dtype): argument
98 index_dtype)
100 def testReduceSumC64(self, index_dtype): argument
102 self.COMPLEX_DATA, index_dtype)
104 def testReduceProdF32(self, index_dtype): argument
106 self.REAL_DATA, index_dtype)
108 def testReduceProdC64(self, index_dtype): argument
110 self.COMPLEX_DATA, index_dtype)
[all …]
|
| D | reshape_op_test.py | 34 def testBasic(self, index_dtype): argument
39 shape = constant_op.constant([3, 2], dtype=index_dtype)
|
| D | listdiff_op_test.py | 35 for index_dtype in [dtypes.int32, dtypes.int64]:
41 x_tensor, y_tensor, out_idx=index_dtype)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_add_op_test.py | 39 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
43 x_indices = np.vstack(non_zero).astype(index_dtype).T
163 for index_dtype in [np.int32, np.int64]:
168 sparse, unused_nnz = _sparsify(rand_vals_np, index_dtype=index_dtype)
222 sp_t, unused_nnz = _sparsify(sp_vals, thresh=sparsity, index_dtype=np.int32)
|
| D | listdiff_op_test.py | 44 for index_dtype in [dtypes.int32, dtypes.int64]:
49 index_dtype=index_dtype)
|
| D | matrix_band_part_op_test.py | 61 for index_dtype in [dtypes_lib.int32, dtypes_lib.int64]:
65 constant_op.constant(lower, index_dtype),
66 constant_op.constant(upper, index_dtype))
|
| D | shape_ops_test.py | 39 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
43 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| D | cwise_ops_unary_test.py | 43 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
47 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| D | cwise_ops_binary_test.py | 48 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
52 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| D | sparse_ops_test.py | 41 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
45 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| D | cwise_ops_test.py | 58 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
62 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op_test.cc | 38 const auto& index_dtype = DataTypeToEnum<INDEX_TYPE>::v(); in BuildRaggedGatherGraph() local
45 .Input(FakeInput(index_dtype)) // indices in BuildRaggedGatherGraph()
49 .Attr("Tindices", index_dtype) in BuildRaggedGatherGraph()
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | adam_test.py | 111 for index_dtype in [dtypes.int32, dtypes.int64]:
116 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | adamax_test.py | 131 for index_dtype in [dtypes.int32, dtypes.int64]:
136 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| D | lazy_adam_optimizer_test.py | 109 for index_dtype in [dtypes.int32, dtypes.int64]:
118 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| D | adam_gs_optimizer_test.py | 117 for index_dtype in [dtypes.int32, dtypes.int64]:
122 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| D | lazy_adam_gs_optimizer_test.py | 114 for index_dtype in [dtypes.int32, dtypes.int64]:
126 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam_test.py | 115 for index_dtype in [dtypes.int32, dtypes.int64]:
120 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adamax_test.py | 133 for index_dtype in [dtypes.int32, dtypes.int64]:
138 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| D | adam_test.py | 160 for index_dtype in [dtypes.int32, dtypes.int64]:
165 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 225 def setdiff1d(x, y, index_dtype=dtypes.int32, name=None): argument
226 return gen_array_ops.list_diff(x, y, index_dtype, name)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2013 …argspec: "args=[\'x\', \'y\', \'index_dtype\', \'name\'], varargs=None, keywords=None, defaults=[\…
|