| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 81 Shape new_shape(old_shape); in RowMajorShape() local
89 Shape new_shape(old_shape); in ColMajorShape() local
|
| /external/XNNPACK/src/subgraph/ |
| D | static-reshape.c | 20 const size_t* new_shape, in xnn_define_static_reshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 46 const Shape& new_shape) { in PadInstruction()
245 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()
354 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
|
| D | variadic_op_splitter.cc | 58 Shape new_shape = concat->shape(); in SplitConcatenate() local
|
| D | cublas_gemm_pad_for_tensor_cores.cc | 67 auto create_padding_config = [](Shape& shape, Shape& new_shape) { in PadForTensorCores()
|
| D | gpu_conv_rewriter.cc | 108 Shape new_shape = lhs->shape(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
600 Shape new_shape = rhs->shape(); in MatchBackwardInput() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_canonicalizer.cc | 36 auto new_shape = ShapeUtil::MakeTupleShape({root_shape}); in CanonicalizeNonTupleConditional() local
|
| D | slice_sinker.cc | 171 Shape new_shape = ShapeUtil::ChangeElementType(shape, element_type); in SinkSlices() local
|
| D | hlo_element_type_converter.cc | 196 Shape new_shape = GetConvertedTupleShape(hlo->shape(), eliminate_type_, in Run() local
|
| D | reshape_mover.cc | 143 const Shape new_shape = in UpdateOperand() 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/llvm-project/flang/test/Evaluate/ |
| D | folding06.f90 | 21 integer(8), parameter :: new_shape(*) = [2, 4] variable
|
| /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 | 129 LogicalResult HandleCast(TF::CastOp cast_op, ArrayRef<int64_t> new_shape) { in HandleCast()
239 TF::ReshapeOp GetReshapeOpForConv2DFilter(ArrayRef<int64_t> new_shape, in GetReshapeOpForConv2DFilter()
293 SmallVector<int64_t, 6> new_shape = { in HandleConv2DFilter() local
359 SmallVector<int64_t, 4> new_shape(new_filter_shape.begin(), in HandleConv2DBackPropFilter() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 253 std::vector<int64> new_shape(existing_dims_size); in Compile() local
286 std::vector<int64> new_shape; in Compile() local
|
| D | dynamic_stitch_op.cc | 154 TensorShape new_shape; in Compile() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 399 HW new_shape; member
412 HWD new_shape; member
521 BHWC new_shape; member
525 BHWDC new_shape; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | cwise_op_vectorizer.cc | 100 Output new_shape = ops::Concat(scope, {concat_pre, ones, concat_post}, in ExpandDimsForBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 256 llvm::SmallVector<int64_t, 2> new_shape{type.getShape()[1], in matchAndRewrite() local
619 auto new_shape = rewriter.create<TF::ConstOp>(op->getLoc(), new_shape_attr); in matchAndRewrite() local
692 auto new_shape = rewriter.create<TF::ConstOp>(op->getLoc(), new_shape_attr); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_nearest_upsample.cc | 52 void CopyShape(const std::vector<int>& new_shape, Array* array) { in CopyShape()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 220 ShapeHandle new_shape; in MergeInput() local
252 ShapeHandle new_shape; in RelaxInput() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.h | 207 std::vector<int64> new_shape; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 283 Shape new_shape = shape; in MakeFakeLiteralInternal() local
385 Shape new_shape = shape; in MakeFakeLiteralInternalWithBounds() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 182 TfLiteIntArray* new_shape = TfLiteIntArrayCopy(output_tensor->dims); in ResizeOutputTensors() local
|