| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_bitcast_lift.cc | 153 Shape new_shape = i->users()[0]->shape(); in Run() local
196 Shape new_shape = i->users()[0]->shape(); in Run() local
211 Shape new_shape = i->users()[0]->shape(); in Run() local
|
| D | cudnn_pad_for_convolutions.cc | 43 const Shape& new_shape) { in PadInstruction()
258 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()
428 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
|
| D | triangular_solve_rewriter.cc | 48 Shape new_shape = ShapeUtil::MakeTupleShape({ in Run() local
|
| D | cudnn_fused_conv_rewriter.cc | 171 Shape new_shape = conv->shape(); in FuseConvertToFloat() local
529 Shape new_shape = conv->shape(); in FuseConvertToF16() local
610 Shape new_shape = conv->shape(); in FuseConvertToS8() local
|
| D | variadic_op_splitter.cc | 58 Shape new_shape = concat->shape(); in SplitConcatenate() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 82 Shape new_shape(old_shape); in RowMajorShape() local
90 Shape new_shape(old_shape); in ColMajorShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convert_mover.cc | 138 Shape new_shape = instr->shape(); in MoveConvertPrecisionOps() local
179 Shape new_shape = to_convert->shape(); in MoveConvertPrecisionOps() local
|
| D | change_op_data_type.cc | 43 Shape new_shape = instr->shape(); in Run() local
|
| D | convolution_pred_expander.cc | 55 Shape new_shape = ShapeUtil::ChangeElementType(instruction->shape(), F16); in ExpandInstruction() local
|
| D | conditional_canonicalizer.cc | 36 auto new_shape = ShapeUtil::MakeTupleShape(absl::MakeSpan(&root_shape, 1)); in CanonicalizeNonTupleConditional() local
|
| D | bitcast_decomposer_test.cc | 48 Shape new_shape = ShapeUtil::MakeShapeWithLayout( in AllPermutationsOfShape() local
|
| D | bitcast_decomposer.cc | 163 Shape new_shape = in Run() local
|
| D | slice_sinker.cc | 171 Shape new_shape = ShapeUtil::ChangeElementType(shape, element_type); in SinkSlices() local
|
| D | hlo_element_type_converter.cc | 205 Shape new_shape = GetConvertedTupleShape(hlo->shape(), eliminate_type_, in Run() local
|
| D | conditional_simplifier.cc | 254 const Shape new_shape = ShapeUtil::MakeTupleShapeWithPtrs(new_tuple_shapes); in RemoveUnusedTupleElements() local
536 Shape new_shape = ShapeUtil::ChangeElementType(shape, PRED); in TryRemoveConditional() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | string_util_test.cc | 70 auto new_shape = TfLiteIntArrayCreate(2); in TEST() local
186 auto new_shape = TfLiteIntArrayCreate(2); in TEST() local
|
| D | string_util.cc | 108 TfLiteIntArray* new_shape) { in WriteToTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 87 auto new_shape = builder->create<TFL::ConstOp>(loc, new_shape_attr); in InsertReshapeOp() local
399 SmallVector<int64_t, 4> new_shape; in matchAndRewrite() local
550 SmallVector<int64_t, 4> new_shape; in matchAndRewrite() local
651 SmallVector<int64_t, 2> new_shape({batch, channel}); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | nullary_spmd_expander.cc | 64 std::vector<int64_t> new_shape(dense.getType().getRank()); in ExpandOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_layout_helper.cc | 67 SmallVector<int64_t, 4> new_shape(permutation.size()); in ShuffleRankedTensorType() local
|
| /external/XNNPACK/src/subgraph/ |
| D | static-reshape.c | 136 const size_t* new_shape, in xnn_define_static_reshape()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.h | 155 void SetBHWCShape(const BHWC& new_shape) { in SetBHWCShape()
158 void SetBHWDCShape(const BHWDC& new_shape) { shape_ = new_shape; } in SetBHWDCShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | legacy_reshape.json | 37 "new_shape": [ 2, 2 ] array
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 84 LogicalResult HandleCast(TF::CastOp cast_op, ArrayRef<int64_t> new_shape) { in HandleCast()
194 TF::ReshapeOp GetReshapeOpForConv2DFilter(ArrayRef<int64_t> new_shape, in GetReshapeOpForConv2DFilter()
248 SmallVector<int64_t, 6> new_shape = { in HandleConv2DFilter() local
314 SmallVector<int64_t, 4> new_shape(new_filter_shape.begin(), in HandleConv2DBackPropFilter() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | tflite_subgraph_execute.cc | 128 std::vector<int> new_shape; in ResizeInputTensor() local
|