- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 233 for tensor_shape (0.15 sec)
-
tensorflow/cc/experimental/libtf/impl/iostream_test.cc
#include "tensorflow/cc/experimental/libtf/impl/scalars.h" #include "tensorflow/cc/experimental/libtf/impl/string.h" #include "tensorflow/cc/experimental/libtf/impl/tensor_spec.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/test.h" namespace tf { namespace libtf { namespace impl { TEST(OStreamTest, TestInt64) { Int64 x(42);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 09:47:46 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/mangling_util.cc
#include "absl/strings/match.h" #include "absl/strings/str_cat.h" #include "tensorflow/compiler/mlir/tensorflow/utils/parse_text_proto.h" #include "tensorflow/core/framework/tensor.pb.h" #include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/ir/importexport/mangling.h" #include "tensorflow/core/lib/core/errors.h" #include "tensorflow/core/platform/protobuf.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-control-ret-same-island.pbtxt
node { name: "const" op: "Const" attr { key: "dtype" value { type: DT_FLOAT } } attr { key: "value" value { tensor { dtype: DT_FLOAT tensor_shape { } float_val: 0.0 } } } } node { name: "test_fn_call" op: "StatefulPartitionedCall" input: "const" attr { key: "Tin" value { list {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 10 19:53:21 UTC 2020 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
ShapedType tensor_shape, const bool is_lhs, SmallVector<int64_t> &opponent_shape) { int opponent_required_dim = xladot_output_rank; int used_rank = tensor_shape.getRank(); if (is_lhs) { used_rank -= dnums.lhs_contracting_dimensions_size(); for (int64_t v : dnums.lhs_contracting_dimensions()) { opponent_shape.push_back(tensor_shape.getDimSize(v)); } } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/function-order.mlir
// CHECK-NEXT: } // CHECK-NEXT: attr { // CHECK-NEXT: key: "value" // CHECK-NEXT: value { // CHECK-NEXT: tensor { // CHECK-NEXT: dtype: DT_INT32 // CHECK-NEXT: tensor_shape { // CHECK-NEXT: } // CHECK-NEXT: int_val: 1 // CHECK-NEXT: } // CHECK-NEXT: } // CHECK-NEXT: } // CHECK: } // CHECK: node_def {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 17 18:52:47 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_tensor_buffer_util.cc
#include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.h" #include "tsl/platform/statusor.h" namespace tensorflow { static size_t GetTensorSize(const TensorShape& shape, const DataType dtype) { return shape.num_elements() * DataTypeSize(dtype); } absl::StatusOr<Tensor> MakeTensorFromPjRtBuffer( const DataType dtype, const TensorShape& shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 14 18:14:47 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/impl/tensor_spec.h
#ifndef TENSORFLOW_CC_EXPERIMENTAL_LIBTF_IMPL_TENSOR_SPEC_H_ #define TENSORFLOW_CC_EXPERIMENTAL_LIBTF_IMPL_TENSOR_SPEC_H_ #include <iosfwd> #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" namespace tf { namespace libtf { namespace impl { /// @brief The TensorSpec struct. /// /// The TensorSpec describes the shape and dtype of a Tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 09 21:11:15 UTC 2021 - 1.7K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/tensor_spec.h
#ifndef TENSORFLOW_C_EXPERIMENTAL_SAVED_MODEL_CORE_TENSOR_SPEC_H_ #define TENSORFLOW_C_EXPERIMENTAL_SAVED_MODEL_CORE_TENSOR_SPEC_H_ #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/protobuf/struct.pb.h" namespace tensorflow { // Note(bmzhao): TensorSpec deliberately does not store the "name" from a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 29 23:11:59 UTC 2020 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.h
#include "mlir/IR/BuiltinTypes.h" // from @llvm-project #include "mlir/IR/Operation.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime #include "tfrt/core_runtime/tensor_handle.h" // from @tf_runtime
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 17:42:41 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc
#include "tensorflow/c/eager/immediate_execution_operation.h" #include "tensorflow/c/eager/immediate_execution_tensor_handle.h" #include "tensorflow/core/framework/resource_handle.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/gtl/inlined_vector.h" #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h" #include "tensorflow/core/platform/errors.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 11:28:19 UTC 2024 - 5K bytes - Viewed (0)