| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_partitioned_output_op.cc | 45 shape_inference::DimensionHandle new_dim; in __anon141003b80102() local
48 true /* evenly_divisible */, &new_dim), in __anon141003b80102()
50 TF_CHECK_OK(c->ReplaceDim(input, partition_dim, new_dim, &newoutput0)); in __anon141003b80102()
|
| 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 | 537 auto new_dim = new_window.add_dimensions(); in HandleConvolution() local
538 new_dim->set_size(depthwise_multiplier); in HandleConvolution()
539 new_dim->set_window_reversal(true); in HandleConvolution()
540 new_dim->set_padding_low(depthwise_multiplier - 1); in HandleConvolution()
541 new_dim->set_padding_high(depthwise_multiplier - 1); in HandleConvolution()
542 new_dim->set_stride(1); in HandleConvolution()
543 new_dim->set_window_dilation(1); in HandleConvolution()
544 new_dim->set_base_dilation(1); in HandleConvolution()
|
| D | algebraic_simplifier.cc | 1979 int64_t new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1980 if (new_dim != -1) { in RemoveDegenerateDimensionFromDot()
1981 new_dnums.add_lhs_batch_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()
1985 int64_t new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1986 if (new_dim != -1) { in RemoveDegenerateDimensionFromDot()
1987 new_dnums.add_lhs_contracting_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()
1992 int64_t new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1993 if (new_dim != -1) { in RemoveDegenerateDimensionFromDot()
1994 new_dnums.add_rhs_batch_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()
1998 int64_t new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | runtime_shape.h | 197 const int new_dim = in ToRuntimeDims() local
199 result.sizes[i] = new_dim; in ToRuntimeDims()
201 cum_prod *= new_dim; in ToRuntimeDims()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_transforms.cc | 121 unsigned new_dim = affine_map.operands.size(); in TileLoop() local
127 builder.getAffineDimExpr(new_dim)); in TileLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 192 WindowDimension* new_dim = new_window.add_dimensions(); in GenNewWindow() local
194 new_dim->set_size(2); in GenNewWindow()
195 new_dim->set_padding_low(0); in GenNewWindow()
196 new_dim->set_padding_high(0); in GenNewWindow()
197 new_dim->set_stride(1); in GenNewWindow()
198 new_dim->set_window_dilation(1); in GenNewWindow()
199 new_dim->set_base_dilation(1); in GenNewWindow()
200 new_dim->set_window_reversal(false); in GenNewWindow()
202 new_dim->set_size(2); in GenNewWindow()
203 new_dim->set_padding_low(0); in GenNewWindow()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 544 int64_t new_dim = orig_dim + low + high; in __anonb720852f0302() local
546 new_dim += interior * (orig_dim - 1); in __anonb720852f0302()
548 if (new_dim < 0) { in __anonb720852f0302()
550 "resulting padded dimension has negative size ", new_dim); in __anonb720852f0302()
552 output_dims.emplace_back(c->MakeDim(new_dim)); in __anonb720852f0302()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 248 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
249 new_dim->set_padding_low(new_conv_padding); in CanonicalizeBackwardFilterConvolution()
250 new_dim->set_padding_high(new_conv_padding); in CanonicalizeBackwardFilterConvolution()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 950 new_dim = tensor_shape.Dimension(None)
952 new_dim = tensor_shape.Dimension(pack_input_val)
953 ret = ret.concatenate([new_dim])
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 661 DimensionHandle new_dim, ShapeHandle* out) { in ReplaceDim() argument
676 dims[dim_index] = new_dim; in ReplaceDim()
|
| D | shape_inference.h | 524 Status ReplaceDim(ShapeHandle s, int64_t 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 | 1195 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); in OutputAsUnion() local
1196 TF_CHECK_OK(ctx->ReplaceDim(relaxed, d, new_dim, &relaxed)); in OutputAsUnion()
|