- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 68 for shape_b (1.46 sec)
-
tensorflow/c/experimental/saved_model/core/tensor_spec.cc
TensorSpec::TensorSpec() : shape_(std::initializer_list<int64_t>()), dtype_(DT_FLOAT) {} TensorSpec::TensorSpec(PartialTensorShape shape, DataType dtype) : shape_(std::move(shape)), dtype_(dtype) {} TensorSpec::TensorSpec(const TensorSpecProto& proto) : shape_(proto.shape()), dtype_(proto.dtype()) {} const PartialTensorShape& TensorSpec::shape() const { return shape_; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 1.3K bytes - Viewed (0) -
test/typeparam/shape1.out
Keith Randall <******@****.***> 1627678827 -0700
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Sat Jul 31 17:03:07 UTC 2021 - 10 bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_device_context.h
se::Stream* stream, se::DeviceMemoryBase* device_memory_base, const xla::Shape& shape, tsl::AsyncValueRef<std::unique_ptr<se::Event>>& done_event) : stream_(stream), device_memory_base_(device_memory_base), shape_(shape), done_event_(done_event) {} // Copies 'cpu_tensor' to `device_memory_base_` with `shape_`. // `device_tensor` is unused.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.h
public: XlaHostRecvDeviceContext( se::Stream* stream, const se::DeviceMemoryBase& device_memory_base, const xla::Shape& shape, tsl::AsyncValueRef<std::unique_ptr<se::Event>>& done_event) : stream_(stream), device_memory_base_(device_memory_base), shape_(shape), done_event_(done_event) {} void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.cc
const Tensor* device_tensor, StringPiece tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) { DataType dtype = EncodePrimitiveTypeAsDataType(shape_.element_type()).value(); TensorShape tensor_shape; Status status = XLAShapeToTensorShape(shape_, &tensor_shape); if (!status.ok()) { done(status); return; } *cpu_tensor = Tensor(dtype, tensor_shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
} // Make invariant the `ShapeOp`s or a `ReadVariableOp` that's the `ShapeOp`'s // predecessor. void MakeShapeOpInvariant(tf_device::ReplicateOp replicate_op, int num_replicas, Block* replicate_block, TF::ShapeOp shape_op) { // Ignore ShapeOps that have virtual devices. if (AncestorUsesVirtualDevice(replicate_op.getDevices(), shape_op)) return; Value input = shape_op.getInput();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/output-shapes.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-unconditionally-use-set-output-shapes-on-import -tf-enable-shape-inference-on-import=false -tf-graph-as-function %s -o - | FileCheck %s # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-unconditionally-use-set-output-shapes-on-import -tf-enable-shape-inference-on-import=true -tf-graph-as-function %s -o - | FileCheck %s # Verify importing with _output_shapes enabled works as expected. node { name: "_Arg"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 21 04:07:13 UTC 2021 - 3K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/revived_types/variable.cc
Variable::Variable(ImmediateExecutionContext* ctx, DataType dtype, TensorShape shape, absl::optional<std::string> name, ImmediateTensorHandlePtr handle) : TensorHandleConvertible(std::move(handle)), name_(name.has_value() ? *name : "Variable"), dtype_(dtype), shape_(shape), ctx_(ctx) {} Variable::~Variable() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 08 20:55:40 UTC 2020 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/tf_to_stablehlo.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 13.6K bytes - Viewed (0)