Searched refs:data_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1003 std::vector<int64> data_dims = RandomDims(); in TEST_F() local
1004 Tensor indices = RandomReductionIndices(data_dims.size()); in TEST_F()
1007 .RandomInput(DT_BOOL, data_dims) in TEST_F()
1023 std::vector<int64> data_dims = RandomDims(); in TEST_F() local
1024 Tensor indices = RandomReductionIndices(data_dims.size()); in TEST_F()
1027 .RandomInput(DT_BOOL, data_dims) in TEST_F()
1604 std::vector<int64> data_dims = in TEST_F() local
1612 .RandomInput(type, data_dims) in TEST_F()
2204 std::vector<int64> data_dims = RandomDims(4, 4, 1, 8); in TEST_F() local
2206 std::uniform_int_distribution<int> radius(1, data_dims[3]); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | bias_op_deterministic_test.py | 44 data_dims = data_rank * (data_dim,)
46 shape = (batch_size,) + (channel_count,) + data_dims
48 shape = (batch_size,) + data_dims + (channel_count,)
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 194 const int data_dims) { in GetDimIndices() argument
198 depth_idx = data_dims - 1; in GetDimIndices()
199 batch_idx = data_dims - 2; in GetDimIndices()
204 depth_idx = data_dims - 2; in GetDimIndices()
205 batch_idx = data_dims - 1; in GetDimIndices()
210 depth_idx = data_dims - 1; in GetDimIndices()
|