Searched refs:expanded_shape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | data_format.cc | 64 std::vector<int64_t> expanded_shape = in Expand() local
66 expanded_shape[dim] /= 4; in Expand()
67 expanded_shape.insert(expanded_shape.begin() + dim + 1, 4); in Expand()
71 const int64_t expanded_shape_size = expanded_shape.size(); in Expand()
81 return xla::Transpose(xla::Reshape(input, expanded_shape), permutation); in Expand()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | spmd_expander.cc | 133 const auto expanded_shape = std::get<0>(expanded_and_true_global_shape); in ExpandOpAndSetLayout() local
137 if (expanded_shape <= 0 || expected_shape <= 0) continue; in ExpandOpAndSetLayout()
139 if (expanded_shape != expected_shape) { in ExpandOpAndSetLayout()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/utils/ |
| D | sparsity_format_converter.cc | 89 std::vector<int> expanded_shape(num_expanded_dims); in DenseToSparse() local
92 expanded_shape[i] = blocked_shape_[i]; in DenseToSparse()
94 expanded_shape[i] = block_size_[i - num_original_dims]; in DenseToSparse()
133 segment_count *= expanded_shape[traversal_order_[i]]; in DenseToSparse()
145 dim_metadata_[i * 2].push_back(expanded_shape[traversal_order_[i]]); in DenseToSparse()
179 int dim_size = expanded_shape[original_dim_idx]; in DenseToSparse()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_fusion.cc | 97 llvm::ArrayRef<int64_t> expanded_shape = in IsUnitDimExpansionOnly() local
103 if (expanded_shape[position] == 1) num_unit_dims++; in IsUnitDimExpansionOnly()
|
| /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_t>(); in Compute()
134 TensorShape expanded_tensor_shape(expanded_shape.vec<int64_t>()); in Compute()
|
| D | sparse_tensors_map_ops.cc | 440 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
443 auto expanded_shape_t = expanded_shape.vec<int64_t>(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 110 Shape expanded_shape = shape; in ExpandedFilterShape() local
111 expanded_shape.set_dimensions( in ExpandedFilterShape()
113 return expanded_shape; in ExpandedFilterShape()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 373 expanded_shape = array_ops.concat([
379 expanded_shape = array_ops.concat(
383 expanded_grad = array_ops.reshape(received_grad, expanded_shape)
|
| D | metrics_impl.py | 3121 expanded_shape = array_ops.concat(
3127 tensor, shape=expanded_shape, name='expand')
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | distribution.py | 1244 expanded_shape = util.pick_vector(
1247 x = array_ops.reshape(x, expanded_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 3286 SmallVector<Attribute, 4> expanded_shape( in ExpandedShape() local
3293 expanded_shape[x.value().getSExtValue()] = in ExpandedShape()
3299 RankedTensorType::get({static_cast<int64_t>(expanded_shape.size())}, in ExpandedShape()
3301 auto attr = DenseElementsAttr::get(attr_type, expanded_shape); in ExpandedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3550 auto expanded_shape = input_hlos[i]->shape(); in HandleReduce() local
3553 expanded_shape.set_dimensions(dim, 1); in HandleReduce()
3558 HloInstruction::CreateReshape(expanded_shape, gte)); in HandleReduce()
|