Home
last modified time | relevance | path

Searched refs:expected_mask (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning_test.py116 def _blockMasking(self, hparams, weights, expected_mask): argument
131 self.assertAllEqual(mask_val, expected_mask)
142 expected_mask = [[0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0],
146 expected_mask)
148 expected_mask)
160 expected_mask = [[[0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0],
164 expected_mask)
166 weights_avg, expected_mask)
/external/deqp-deps/SPIRV-Tools/test/
Dtext_to_binary.memory_test.cpp59 const uint32_t expected_mask = in TEST_F() local
61 EXPECT_THAT(expected_mask, Eq(3u)); in TEST_F()
63 Eq(MakeInstruction(SpvOpStore, {1, 2, expected_mask, 16}))); in TEST_F()
Dtext_to_binary.control_flow_test.cpp65 const uint32_t expected_mask = in TEST_F() local
68 Eq(MakeInstruction(SpvOpSelectionMerge, {1, expected_mask}))); in TEST_F()
125 const uint32_t expected_mask = in TEST_F() local
128 Eq(MakeInstruction(SpvOpLoopMerge, {1, 2, expected_mask}))); in TEST_F()
Dtext_to_binary.function_test.cpp64 const uint32_t expected_mask = SpvFunctionControlInlineMask | in TEST_F() local
68 Eq(MakeInstruction(SpvOpFunction, {1, 2, expected_mask, 3}))); in TEST_F()
Dtext_to_binary.annotation_test.cpp288 const uint32_t expected_mask = SpvFPFastMathModeNotNaNMask | in TEST_F() local
293 Eq(MakeInstruction(SpvOpDecorate, {1, expected_enum, expected_mask}))); in TEST_F()
/external/swiftshader/third_party/SPIRV-Tools/test/
Dtext_to_binary.memory_test.cpp59 const uint32_t expected_mask = in TEST_F() local
61 EXPECT_THAT(expected_mask, Eq(3u)); in TEST_F()
63 Eq(MakeInstruction(SpvOpStore, {1, 2, expected_mask, 16}))); in TEST_F()
Dtext_to_binary.control_flow_test.cpp65 const uint32_t expected_mask = in TEST_F() local
68 Eq(MakeInstruction(SpvOpSelectionMerge, {1, expected_mask}))); in TEST_F()
125 const uint32_t expected_mask = in TEST_F() local
128 Eq(MakeInstruction(SpvOpLoopMerge, {1, 2, expected_mask}))); in TEST_F()
Dtext_to_binary.function_test.cpp64 const uint32_t expected_mask = SpvFunctionControlInlineMask | in TEST_F() local
68 Eq(MakeInstruction(SpvOpFunction, {1, 2, expected_mask, 3}))); in TEST_F()
Dtext_to_binary.annotation_test.cpp288 const uint32_t expected_mask = SpvFPFastMathModeNotNaNMask | in TEST_F() local
293 Eq(MakeInstruction(SpvOpDecorate, {1, expected_enum, expected_mask}))); in TEST_F()
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/
Dmasked_autoregressive_test.py39 expected_mask = np.array(
47 self.assertAllEqual(expected_mask, mask)
50 expected_mask = np.array(
58 self.assertAllEqual(expected_mask, mask)