Searched defs:reshape_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_tester.cc | 127 const std::array<int32_t, 1> reshape_shape{ in CreateTfLiteModel() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | optimize.cc | 62 ArrayRef<int64_t> reshape_shape = reshape_type.getShape(); in matchAndRewrite() local
|
| D | einsum.cc | 330 std::vector<int64_t> reshape_shape = in rewriteToBatchMatmul() local
|
| D | legalize_hlo.cc | 743 std::vector<int64_t> reshape_shape(type.getRank(), 1); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | rsqrt_builder.cc | 108 const int reshape_shape[] = {1, 1, 1, 4}; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 339 Shape reshape_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 606 const std::string reshape_shape = in ConvertFullyConnectedOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 244 llvm::SmallVector<int64_t, 8> reshape_shape; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 2521 Shape reshape_shape = builder in TEST_F() local
4320 Shape reshape_shape = ShapeUtil::MakeShape(F32, {30, 1, 4}); in TEST_F() local
|