| /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/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 37 const Shape& new_shape) { in PadInstruction()
248 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()
411 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
|
| D | variadic_op_splitter.cc | 58 Shape new_shape = concat->shape(); in SplitConcatenate() local
|
| D | gpu_conv_rewriter.cc | 108 Shape new_shape = lhs->shape(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
627 Shape new_shape = rhs->shape(); in MatchBackwardInput() local
|
| D | cublas_pad_for_gemms.cc | 68 auto create_padding_config = [](Shape& shape, Shape& new_shape) { in PadForGemm()
|
| /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/xla/service/ |
| D | conditional_canonicalizer.cc | 36 auto new_shape = ShapeUtil::MakeTupleShape({root_shape}); in CanonicalizeNonTupleConditional() local
|
| D | conditional_simplifier.cc | 252 const Shape new_shape = ShapeUtil::MakeTupleShape(new_tuple_shapes); in RemoveUnusedTupleElements() local
534 Shape new_shape = ShapeUtil::ChangeElementType(shape, PRED); in TryRemoveConditional() local
|
| D | slice_sinker.cc | 171 Shape new_shape = ShapeUtil::ChangeElementType(shape, element_type); in SinkSlices() local
|
| D | hlo_element_type_converter.cc | 199 Shape new_shape = GetConvertedTupleShape(hlo->shape(), eliminate_type_, in Run() local
|
| D | reshape_mover.cc | 143 const Shape new_shape = in UpdateOperand() local
|
| /external/XNNPACK/src/subgraph/ |
| D | static-reshape.c | 132 const size_t* new_shape, in xnn_define_static_reshape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 83 LogicalResult HandleCast(TF::CastOp cast_op, ArrayRef<int64_t> new_shape) { in HandleCast()
193 TF::ReshapeOp GetReshapeOpForConv2DFilter(ArrayRef<int64_t> new_shape, in GetReshapeOpForConv2DFilter()
247 SmallVector<int64_t, 6> new_shape = { in HandleConv2DFilter() local
313 SmallVector<int64_t, 4> new_shape(new_filter_shape.begin(), in HandleConv2DBackPropFilter() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | legacy_reshape.json | 37 "new_shape": [ 2, 2 ] array
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 83 auto new_shape = builder->create<TFL::ConstOp>(loc, new_shape_attr); in InsertReshapeOp() local
395 SmallVector<int64_t, 4> new_shape; in matchAndRewrite() local
546 SmallVector<int64_t, 4> new_shape; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | tflite_subgraph_execute.cc | 117 std::vector<int> new_shape; in ResizeInputTensor() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 281 std::vector<int64> new_shape(existing_dims_size); in Compile() local
314 std::vector<int64> new_shape; in Compile() local
|
| D | dynamic_stitch_op.cc | 164 TensorShape new_shape; in Compile() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | OpticalFlowFixture.h | 85 TensorShape new_shape = library->get_image_shape(new_image_name); in compute_target() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 408 HW new_shape; member
421 HWD new_shape; member
544 BHWC new_shape; member
548 BHWDC new_shape; member
|
| /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/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 292 llvm::SmallVector<int64_t, 2> new_shape{type.getShape()[1], in matchAndRewrite() local
746 auto new_shape = rewriter.create<TF::ConstOp>(op->getLoc(), new_shape_attr); in matchAndRewrite() local
884 auto new_shape = rewriter.create<TF::ConstOp>(op->getLoc(), new_shape_attr); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 222 ShapeHandle new_shape; in MergeInput() local
254 ShapeHandle new_shape; in RelaxInput() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.h | 207 std::vector<int64> new_shape; in Compute() local
|