| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_ops_test.py | 44 index_dtype, argument
53 index = array_ops.placeholder(index_dtype)
101 def testReduceSumF32(self, index_dtype): argument
103 index_dtype)
105 def testReduceSumC64(self, index_dtype): argument
107 self.COMPLEX_DATA, index_dtype)
109 def testReduceProdF32(self, index_dtype): argument
111 self.REAL_DATA, index_dtype)
113 def testReduceProdC64(self, index_dtype): argument
115 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/keras/preprocessing/ |
| D | timeseries.py | 183 index_dtype = 'int32'
185 index_dtype = 'int64'
188 start_positions = np.arange(0, num_seqs, sequence_stride, dtype=index_dtype)
195 sequence_length = math_ops.cast(sequence_length, dtype=index_dtype)
196 sampling_rate = math_ops.cast(sampling_rate, dtype=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 | segment_reduction_ops_test.py | 549 for index_dtype in index_dtypes:
563 indices=math_ops.cast(tf_indices, index_dtype),
|
| D | cwise_ops_unary_test.py | 46 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
50 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_binary_test.py | 49 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
53 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| D | cwise_ops_test.py | 56 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
60 x_indices = np.vstack(non_zero).astype(index_dtype).T
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op_test.cc | 40 const auto& index_dtype = DataTypeToEnum<INDEX_TYPE>::v(); in BuildRaggedGatherGraph() local
47 .Input(FakeInput(index_dtype)) // indices in BuildRaggedGatherGraph()
51 .Attr("Tindices", index_dtype) in BuildRaggedGatherGraph()
|
| D | ragged_tensor_to_tensor_op_test.cc | 59 const auto& index_dtype = DataTypeToEnum<INDEX_TYPE>::v(); in BuildRaggedTensorToTensorGraph() local
64 .Attr("Tindex", index_dtype) in BuildRaggedTensorToTensorGraph()
67 .Input(FakeInput(index_dtype)) in BuildRaggedTensorToTensorGraph()
71 index_dtype)) // row_partition_tensors in BuildRaggedTensorToTensorGraph()
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adamax_test.py | 130 for index_dtype in [dtypes.int32, dtypes.int64]:
136 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| D | adam_test.py | 160 for index_dtype in [dtypes.int32, dtypes.int64]:
166 indices = constant_op.constant([0, 1], dtype=index_dtype)
612 for index_dtype in [dtypes.int32, dtypes.int64]:
618 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam_test.py | 116 for index_dtype in [dtypes.int32, dtypes.int64]:
121 indices = constant_op.constant([0, 1], dtype=index_dtype)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 462 def setdiff1d(x, y, index_dtype=dtypes.int32, name=None): argument
498 return gen_array_ops.list_diff(x, y, index_dtype, name)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2097 …argspec: "args=[\'x\', \'y\', \'index_dtype\', \'name\'], varargs=None, keywords=None, defaults=[\…
|