Searched defs:index_dtype (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_ops_test.py | 42 index_dtype, argument
96 def testReduceSumF32(self, index_dtype): argument
100 def testReduceSumC64(self, index_dtype): argument
104 def testReduceProdF32(self, index_dtype): argument
108 def testReduceProdC64(self, index_dtype): argument
112 def testReduceMin(self, index_dtype): argument
129 def testReduceMax(self, index_dtype): argument
147 def testReduceMeanF32(self, index_dtype): argument
153 def testReduceMeanF16(self, index_dtype): argument
158 def testReduceMeanC64(self, index_dtype): argument
[all …]
|
| D | reshape_op_test.py | 34 def testBasic(self, index_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_add_op_test.py | 39 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
|
| D | cwise_ops_unary_test.py | 43 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
|
| D | shape_ops_test.py | 39 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
|
| D | sparse_ops_test.py | 41 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
|
| D | cwise_ops_binary_test.py | 48 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
|
| D | cwise_ops_test.py | 58 def _sparsify(x, thresh=0.5, index_dtype=np.int64): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op_test.cc | 38 const auto& index_dtype = DataTypeToEnum<INDEX_TYPE>::v(); in BuildRaggedGatherGraph() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 225 def setdiff1d(x, y, index_dtype=dtypes.int32, name=None): argument
|