- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 639 for ShapeN (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/prepare_tpu_computation_for_tf_export.mlir
// CHECK-SAME: key = "" // CHECK-SAME: recv_key = "host_compute_channel_recv" // CHECK-SAME: send_key = "host_compute_channel_send" // CHECK-SAME: shape_inference_graph = @host_func // CHECK-SAME: shapes = [#tf_type.shape<*>, #tf_type.shape<3x?>] // CHECK-SAME: tpu_core = 0 : i64 // CHECK: func @host_func // CHECK: %[[RECV_OUTPUT:[0-9]*]]:2 = "tf._XlaRecvAtHost" // CHECK-SAME: key = "host_compute_channel_send"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 18:46:36 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/multi-output-feeds.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-prune-unused-nodes -tf-input-arrays=z:1,z:2 -tf-input-shapes=':' -tf-output-arrays=z:2,z:1,a:0 -o - | FileCheck --check-prefix=PRUNE %s
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 10 19:53:21 UTC 2020 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/graph.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-as-function | tf-mlir-translate -mlir-tf-graph-to-hlo-text -tf-input-shapes='' -tf-input-data-types=DT_FLOAT -tf-xla-emit-return-tuple | FileCheck %s node { name: "arg" op: "_Arg" attr { key: "T" value { type: DT_FLOAT } } attr { key: "index" value { i: 0 } } } node {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Oct 16 02:25:37 UTC 2021 - 963 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/shape_list_attr.mlir
// CHECK-NEXT: type: DT_FLOAT // CHECK-NEXT: type: DT_INT16 // CHECK: attr { // CHECK-NEXT: key: "shapes" // CHECK-NEXT: value { // CHECK-NEXT: list { // CHECK-NEXT: shape { // CHECK-NEXT: dim { // CHECK-NEXT: size: 3 // CHECK: shape { // CHECK-NEXT: dim { // CHECK-NEXT: size: 4 // CHECK-NEXT: } // CHECK-NEXT: dim {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 931 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/shapes_for_arguments.py
class TestModule(tf.Module): # Check that we get shapes annotated on function arguments. # # Besides checking the shape on the function input argument, this test also # checks that the shape on the input argument is propagated to the return # value. # We eventually want to move the shape inference to a pass separate from
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
// in a `shape` shaped tensor. Assumes row-major order. `indices` and `shape` // should have the same size. // Example: Index (2, 3) of a (4, 5)-shaped tensor has the contiguous offset of // 2 * 5 + 3 = 13. int64_t GetContiguousOffset(const ArrayRef<int64_t> indices, const ArrayRef<int64_t> shape) { int64_t contiguous_offset = 0; int64_t base_offset = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_type.h
// is converted directly. If it is a shaped type, the element type is converted. Status ConvertToDataType(mlir::Type type, DataType* dtype); // Converts an TensorFlow shape to the one used in MLIR. void ConvertToMlirShape(const TensorShape& input_shape, llvm::SmallVectorImpl<int64_t>* shape); // Converts an TensorFlow shape proto to the one used in MLIR.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/unroll_batch_matmul.pbtxt
node { name: "Placeholder" op: "Placeholder" attr { key: "dtype" value { type: DT_FLOAT } } attr { key: "shape" value { shape { dim { size: 2 }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.h
namespace tensorflow { llvm::SmallVector<int64_t> ConvertTFShapeToMlir(llvm::ArrayRef<int64_t> shapes); llvm::SmallVector<int64_t> ConvertMlirShapeToTF(llvm::ArrayRef<int64_t> shape); static constexpr int64_t kTFDynamicSize = -1; mlir::RankedTensorType GetTypeFromTFTensorShape(llvm::ArrayRef<int64_t> shape, mlir::Type elementType,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 18 11:40:17 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h
absl::string_view shapes, GraphImportConfig::InputArrays* inputs); Status ParseInputArrayInfo( const std::vector<string>& node_names, const std::vector<string>& node_dtypes, const std::vector<std::optional<std::vector<int>>>& node_shapes, GraphImportConfig::InputArrays* inputs); // Parses shapes from the given string into shapes_vector which is a structured
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 04:56:10 UTC 2024 - 6.4K bytes - Viewed (0)