Searched refs:expanded_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | data_format.cc | 60 std::vector<int64> expanded_shape(input_shape.dimensions()); in Expand() local
61 expanded_shape[dim] /= 4; in Expand()
62 expanded_shape.insert(expanded_shape.begin() + dim, 4); in Expand()
66 for (int64 i = 0; i != expanded_shape.size(); ++i) { in Expand()
74 return xla::Transpose(xla::Reshape(input, expanded_shape), permutation); in Expand()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_string_op.cc | 126 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
128 auto expanded_shape_t = expanded_shape.vec<int64>(); in Compute()
132 TensorShape expanded_tensor_shape(expanded_shape.vec<int64>()); in Compute()
|
| D | sparse_tensors_map_ops.cc | 428 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
431 auto expanded_shape_t = expanded_shape.vec<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning_utils.py | 145 expanded_shape = [tensor_shape[0] * multiple, tensor_shape[1]]
148 expanded_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 108 Shape expanded_shape = shape; in ExpandedFilterShape() local
109 expanded_shape.set_dimensions( in ExpandedFilterShape()
111 return expanded_shape; in ExpandedFilterShape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 49 xla::Shape expanded_shape = shape; in ExpandedFilterShapeForDepthwiseConvolution() local
50 expanded_shape.set_dimensions( in ExpandedFilterShapeForDepthwiseConvolution()
53 return expanded_shape; in ExpandedFilterShapeForDepthwiseConvolution()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 367 expanded_shape = array_ops.concat([
373 expanded_shape = array_ops.concat(
377 expanded_grad = array_ops.reshape(received_grad, expanded_shape)
|
| D | metrics_impl.py | 2914 expanded_shape = array_ops.concat(
2920 tensor, shape=expanded_shape, name='expand')
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | distribution.py | 1251 expanded_shape = util.pick_vector(
1254 x = array_ops.reshape(x, expanded_shape)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 577 expanded_shape = [self.virtual_batch_size, -1] + original_shape[1:]
580 inputs = array_ops.reshape(inputs, expanded_shape)
|