Searched refs:data_dims (Results 1 – 2 of 2) sorted by relevance
1002 std::vector<int64> data_dims = RandomDims(); in TEST_F() local1003 Tensor indices = RandomReductionIndices(data_dims.size()); in TEST_F()1006 .RandomInput(DT_BOOL, data_dims) in TEST_F()1022 std::vector<int64> data_dims = RandomDims(); in TEST_F() local1023 Tensor indices = RandomReductionIndices(data_dims.size()); in TEST_F()1026 .RandomInput(DT_BOOL, data_dims) in TEST_F()1603 std::vector<int64> data_dims = in TEST_F() local1611 .RandomInput(type, data_dims) in TEST_F()2203 std::vector<int64> data_dims = RandomDims(4, 4, 1, 8); in TEST_F() local2205 std::uniform_int_distribution<int> radius(1, data_dims[3]); in TEST_F()[all …]
163 const int data_dims) { in GetDimIndices() argument167 depth_idx = data_dims - 1; in GetDimIndices()168 batch_idx = data_dims - 2; in GetDimIndices()173 depth_idx = data_dims - 2; in GetDimIndices()174 batch_idx = data_dims - 1; in GetDimIndices()179 depth_idx = data_dims - 1; in GetDimIndices()