Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SetShapeAttribute (0.13 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

    template <typename ContainerT,
              typename = typename std::enable_if<std::is_same<
                  std::optional<llvm::ArrayRef<int64_t>>,
                  decltype(*std::declval<ContainerT>().begin())>::value>::type>
    void SetShapeAttribute(absl::string_view name, ContainerT shapes,
                           AttrValueMap* values) {
      AttrValue value;
      auto& shape_list = *value.mutable_list();
      for (const std::optional<llvm::ArrayRef<int64_t>>& shape : shapes) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/export_utils.h

        }
      } else {
        proto->set_unknown_rank(true);
      }
    }
    
    // Sets shape attribute with the given name. If the attribute already exists
    // with a different value, returns an error.
    Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shape,
                             AttrValueMap* values);
    
    // Returns true if the given instruction is an mlir::TF::LegacyCallOp or the
    // result of such an operation transformed by the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

              std::move(value);
        }
      }
      for (auto& it : func_call_attrs) {
        (*values)[it.first] = std::move(it.second);
      }
      return absl::OkStatus();
    }
    
    Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shaped_type,
                             AttrValueMap* values) {
      AttrValue value;
      SetTensorShapeProto(shaped_type, value.mutable_list()->add_shape());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

          }
    
          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

          }
    
          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
Back to top