Searched refs:vec_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 245 vec_shape = vec.get_shape().as_list()
247 vec_leading_shape = vec_shape[:-self.block_depth]
249 vec_block_shape = vec_shape[-self.block_depth:]
253 vec_shape = array_ops.shape(vec)
254 vec_leading_shape = vec_shape[:-self.block_depth]
255 vec_block_shape = vec_shape[-self.block_depth:]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1199 auto vec_shape = ShapeUtil::MakeShape(F32, {42}); in TEST_F() local
1213 HloInstruction::CreateParameter(0, vec_shape, "vec_param")); in TEST_F()
1215 HloInstruction::CreateUnary(vec_shape, HloOpcode::kExp, call_param)); in TEST_F()
1222 HloInstruction::CreateParameter(0, vec_shape, "param")); in TEST_F()
1224 HloInstruction::CreateCall(vec_shape, {param}, call_computation)); in TEST_F()
1226 HloInstruction::CreateMap(vec_shape, {call}, map_computation)); in TEST_F()
|