Searched refs:expanded_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | data_format.cc | 62 std::vector<int64> expanded_shape = in Expand() local
64 expanded_shape[dim] /= 4; in Expand()
65 expanded_shape.insert(expanded_shape.begin() + dim + 1, 4); in Expand()
69 for (int64 i = 0, end = expanded_shape.size(); i != end; ++i) { in Expand()
77 return xla::Transpose(xla::Reshape(input, expanded_shape), permutation); in Expand()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 87 std::vector<int> expanded_shape(num_expanded_dims); in DenseToSparse() local
90 expanded_shape[i] = blocked_shape_[i]; in DenseToSparse()
92 expanded_shape[i] = block_size_[i - num_original_dims]; in DenseToSparse()
131 segment_count *= expanded_shape[traversal_order_[i]]; in DenseToSparse()
143 dim_metadata_[i * 2].push_back(expanded_shape[traversal_order_[i]]); in DenseToSparse()
177 int dim_size = expanded_shape[original_dim_idx]; in DenseToSparse()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_string_op.cc | 128 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
130 auto expanded_shape_t = expanded_shape.vec<int64>(); in Compute()
134 TensorShape expanded_tensor_shape(expanded_shape.vec<int64>()); in Compute()
|
| D | sparse_tensors_map_ops.cc | 429 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
432 auto expanded_shape_t = expanded_shape.vec<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 107 Shape expanded_shape = shape; in ExpandedFilterShape() local
108 expanded_shape.set_dimensions( in ExpandedFilterShape()
110 return expanded_shape; in ExpandedFilterShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1207 SmallVector<Attribute, 4> expanded_shape( in ExpandedShape() local
1214 expanded_shape[x.value().getSExtValue()] = in ExpandedShape()
1220 RankedTensorType::get({static_cast<int64_t>(expanded_shape.size())}, in ExpandedShape()
1222 auto attr = DenseElementsAttr::get(attr_type, expanded_shape); in ExpandedShape()
|
| /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/ops/ |
| D | nn_grad.py | 377 expanded_shape = array_ops.concat([
383 expanded_shape = array_ops.concat(
387 expanded_grad = array_ops.reshape(received_grad, expanded_shape)
|
| D | metrics_impl.py | 2942 expanded_shape = array_ops.concat(
2948 tensor, shape=expanded_shape, name='expand')
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 749 expanded_shape = array_ops.concat([
756 inputs = array_ops.reshape(inputs, expanded_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2779 auto expanded_shape = input_hlos[i]->shape(); in HandleReduce() local
2782 expanded_shape.set_dimensions(dim, 1); in HandleReduce()
2787 HloInstruction::CreateReshape(expanded_shape, gte)); in HandleReduce()
|