Searched refs:vec_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 244 vec_shape = vec.shape.as_list()
246 vec_leading_shape = vec_shape[:-self.block_depth]
248 vec_block_shape = vec_shape[-self.block_depth:]
252 vec_shape = array_ops.shape(vec)
253 vec_leading_shape = vec_shape[:-self.block_depth]
254 vec_block_shape = vec_shape[-self.block_depth:]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1371 auto vec_shape = ShapeUtil::MakeShape(F32, {42}); in TEST_F() local
1385 HloInstruction::CreateParameter(0, vec_shape, "vec_param")); in TEST_F()
1387 HloInstruction::CreateUnary(vec_shape, HloOpcode::kExp, call_param)); in TEST_F()
1394 HloInstruction::CreateParameter(0, vec_shape, "param")); in TEST_F()
1396 HloInstruction::CreateCall(vec_shape, {param}, call_computation)); in TEST_F()
1398 HloInstruction::CreateMap(vec_shape, {call}, map_computation)); in TEST_F()
|