| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_partitioned_output_op.cc | 45 shape_inference::DimensionHandle new_dim; in __anonea6454330102() local
48 true /* evenly_divisible */, &new_dim), in __anonea6454330102()
50 TF_CHECK_OK(c->ReplaceDim(input, partition_dim, new_dim, &newoutput0)); in __anonea6454330102()
|
| D | tpu_partitioned_input_op.cc | 29 shape_inference::DimensionHandle new_dim; in _UpdatePartitionDim() local
31 c->Multiply(c->Dim(handle, partition_dim), c->num_inputs(), &new_dim)); in _UpdatePartitionDim()
32 TF_CHECK_OK(c->ReplaceDim(handle, partition_dim, new_dim, &newoutput0)); in _UpdatePartitionDim()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 531 auto new_dim = new_window.add_dimensions(); in HandleConvolution() local
532 new_dim->set_size(depthwise_multiplier); in HandleConvolution()
533 new_dim->set_window_reversal(true); in HandleConvolution()
534 new_dim->set_padding_low(depthwise_multiplier - 1); in HandleConvolution()
535 new_dim->set_padding_high(depthwise_multiplier - 1); in HandleConvolution()
536 new_dim->set_stride(1); in HandleConvolution()
537 new_dim->set_window_dilation(1); in HandleConvolution()
538 new_dim->set_base_dilation(1); in HandleConvolution()
|
| D | algebraic_simplifier.cc | 1765 int64 new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1766 if (new_dim != -1) { in RemoveDegenerateDimensionFromDot()
1767 new_dnums.add_lhs_batch_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()
1771 int64 new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1772 if (new_dim != -1) { in RemoveDegenerateDimensionFromDot()
1773 new_dnums.add_lhs_contracting_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()
1778 int64 new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1779 if (new_dim != -1) { in RemoveDegenerateDimensionFromDot()
1780 new_dnums.add_rhs_batch_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()
1784 int64 new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
[all …]
|
| D | hlo_sharding_util.cc | 93 int64 new_dim = to_merge->tile_assignment().dim(i); in MergeSharding() local
95 if (new_dim == 1) { in MergeSharding()
98 new_tile_dims.push_back(new_dim); in MergeSharding()
99 } else if (new_dim == old_dim) { in MergeSharding()
100 new_tile_dims.push_back(new_dim); in MergeSharding()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 179 WindowDimension* new_dim = new_window.add_dimensions(); in GenNewWindow() local
181 new_dim->set_size(2); in GenNewWindow()
182 new_dim->set_padding_low(0); in GenNewWindow()
183 new_dim->set_padding_high(0); in GenNewWindow()
184 new_dim->set_stride(1); in GenNewWindow()
185 new_dim->set_window_dilation(1); in GenNewWindow()
186 new_dim->set_base_dilation(1); in GenNewWindow()
187 new_dim->set_window_reversal(false); in GenNewWindow()
189 new_dim->set_size(2); in GenNewWindow()
190 new_dim->set_padding_low(0); in GenNewWindow()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 468 int64 new_dim = orig_dim + low + high; in __anon3500048a0302() local
470 new_dim += interior * (orig_dim - 1); in __anon3500048a0302()
472 if (new_dim < 0) { in __anon3500048a0302()
474 "resulting padded dimension has negative size ", new_dim); in __anon3500048a0302()
476 output_dims.emplace_back(c->MakeDim(new_dim)); in __anon3500048a0302()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_transforms.cc | 119 unsigned new_dim = affine_map.operands.size(); in TileLoop() local
125 builder.getAffineDimExpr(new_dim)); in TileLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 245 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
246 new_dim->set_padding_low(new_conv_padding); in CanonicalizeBackwardFilterConvolution()
247 new_dim->set_padding_high(new_conv_padding); in CanonicalizeBackwardFilterConvolution()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 495 const int new_dim =
497 result.sizes[i] = new_dim;
499 cum_prod *= new_dim;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 332 const int new_dim = in ToRuntimeDims() local
334 result.sizes[i] = new_dim; in ToRuntimeDims()
336 cum_prod *= new_dim; in ToRuntimeDims()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 951 new_dim = tensor_shape.Dimension(None)
953 new_dim = tensor_shape.Dimension(pack_input_val)
954 ret = ret.concatenate([new_dim])
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 654 DimensionHandle new_dim, ShapeHandle* out) { in ReplaceDim() argument
669 dims[dim_index] = new_dim; in ReplaceDim()
|
| D | shape_inference.h | 439 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| D | common_shape_fns.cc | 235 DimensionHandle new_dim = c->RankKnown(input_shape) in EinsumShape() local
243 c->Merge(label_to_dimension[label], new_dim, &merged)); in EinsumShape()
246 label_to_dimension[label] = new_dim; in EinsumShape()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1158 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); in OutputAsUnion() local
1159 TF_CHECK_OK(ctx->ReplaceDim(relaxed, d, new_dim, &relaxed)); in OutputAsUnion()
|