Searched defs:reshape_shape (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | reshape_tester.cc | 127 const std::array<int32_t, 1> reshape_shape{ in CreateTfLiteModel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bitcast_decomposer.cc | 198 Shape reshape_shape = ShapeUtil::MakeShapeWithDescendingLayout( in Run() local
|
| H A D | dynamic_padder.cc | 1478 Shape reshape_shape = ShapeUtil::DeleteDimension(i, slice_shape); in RewriteDynamicBinaryOp() local
|
| H A D | algebraic_simplifier_test.cc | 2936 Shape reshape_shape = builder in TEST_F() local
4756 Shape reshape_shape = ShapeUtil::MakeShape(F32, {30, 1, 4}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | optimize.cc | 64 ArrayRef<int64_t> reshape_shape = reshape_type.getShape(); in matchAndRewrite() local
|
| H A D | einsum.cc | 657 std::vector<int64_t> reshape_shape = in rewriteToBatchMatmul() local
|
| H A D | legalize_hlo.cc | 2269 std::vector<int64_t> reshape_shape(type.getRank(), 1); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | rsqrt_builder.cc | 108 const int reshape_shape[] = {1, 1, 1, 4}; in PopulateSubGraph() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_ssd_mobilenet.cpp | 440 … const std::pair<QuantizationInfo, QuantizationInfo> quant_info, TensorShape reshape_shape) in get_node_C_qasymm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_instruction_fusion_test.cc | 321 Shape reshape_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | optimize.cc | 1424 auto reshape_shape = reshape_op.getType().cast<ShapedType>(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | export_tensorflow.cc | 611 const std::string reshape_shape = in ConvertFullyConnectedOperator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_common.cc | 2613 llvm::SmallVector<int64_t> reshape_shape; in convertGenericReduceOp() local
|