Searched refs:z_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops_test.py | 42 self.z_size = 4
45 tensor = math_ops.range(0, self.x_size * self.channel_size * self.z_size *
48 tensor, [self.x_size, self.channel_size, self.z_size, self.probs_size])
58 self.a2_resolved = ('z', self.z_size)
252 new_dim_size = self.channel_size * self.z_size * self.probs_size
278 new_dim_size = self.channel_size * self.z_size * self.probs_size
|
| D | core_test.py | 264 self.z_size = 4
267 tensor = math_ops.range(0, self.x_size * self.channel_size * self.z_size *
270 tensor, [self.x_size, self.channel_size, self.z_size, self.probs_size])
403 1, self.x_size, 1, self.channel_size, self.z_size, self.probs_size,
420 [self.x_size, self.channel_size, 1, self.z_size, self.probs_size]),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 936 const int64 z_size = ShapeUtil::GetDimension(lhs_shape, input_z_dim); in HandleConvolution() local
962 for (int64 iz = 0; iz < z_size; ++iz) { in HandleConvolution()
|