- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 1,131 for shake (0.05 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/mlir_passthrough_op.pbtxt
attr { key: "_user_specified_name" value { s: "x" } } attr { key: "dtype" value { type: DT_FLOAT } } attr { key: "shape" value { shape { dim { size: 10 } } } } } node { name: "y" op: "Placeholder" attr { key: "_user_specified_name" value { s: "y"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_tensor_buffer_util.cc
#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, std::unique_ptr<xla::PjRtBuffer> pjrt_buffer) { TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::PjRtBuffer::ExternalReference> ref,
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/compiler/jit/pjrt_device_context.cc
cpu_tensor->shape(), cpu_tensor->dtype(), std::nullopt); TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns.shape_representation_fn( cpu_tensor->shape(), cpu_tensor->dtype(), /*fast_mem=*/false, layout_preference)); const xla::Layout* device_layout = &(shape.layout()); // The device id should match the local_hardware_id in
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-multiple-while-body-func.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
.getOutputDimensions() .getDefiningOp<shape::ShapeOfOp>(); // Check if the shape come from the original conv op. if (!shape_of_op || shape_of_op.getArg().getDefiningOp<mhlo::ConvolutionOp>() != conv_op) { return failure(); } Value new_shape_of = rewriter.create<shape::ShapeOfOp>( mul_op.getLoc(), shape_of_op.getType(), new_conv);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.h
// xla::Shape shape(xla::F32, {2, 2}, {}, {}) // tsl::AsyncValueRef<std::unique_ptr<se::Event>> done_event = // tsl::MakeConstructedAsyncValueRef<std::unique_ptr<se::Event>>(stream.parent()); // done_event->Init(); // Tensor dest_cpu_tensor; // // XlaHostRecvDeviceContext device_context(&stream, gpu_dst, // shape, done_event); // device_context.CopyDeviceTensorToCPUSync(
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/utils/convert_type.cc
llvm::SmallVectorImpl<int64_t>* shape) { shape->reserve(input_shape.dims()); for (const auto& d : input_shape) { shape->push_back(d.size == kTFDynamicSize ? ShapedType::kDynamic : d.size); } } Status ConvertToMlirShape(const TensorShapeProto& input_shape, llvm::SmallVectorImpl<int64_t>* shape) { shape->reserve(input_shape.dim_size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
} // Helper method that given 'shape' and 'current_index' representing // index in broadcasted tensor, get the index in the flat original tensor. // 'shape' is computed from the original shape and the broadcast dimensions to // match result shape. int64_t GetElementIndex(llvm::SmallVectorImpl<int64_t> &shape, llvm::SmallVectorImpl<int64_t> ¤t_index) { int64_t ind = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc
TF_RETURN_IF_ERROR(varhandle_op->SetAttrType("dtype", dtype)); // Note that if shape is unknown rank, shape.dim_sizes() will be empty, and // shape.dims() will be -1. absl::InlinedVector<int64_t, 4UL> dim_sizes = shape.dim_sizes(); TF_RETURN_IF_ERROR(varhandle_op->SetAttrShape( "shape", reinterpret_cast<const int64_t*>(dim_sizes.data()), shape.dims())); TF_RETURN_IF_ERROR(varhandle_op->SetAttrString("container", "", 0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 11:28:19 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
// 'shape' is the original shape with padding to match result shape. int64_t GetElementIndex(const std::vector<int64_t>& shape, const std::vector<int64_t>& current_index) { int64_t ind = 0; int64_t mul = 1; for (int i = shape.size() - 1; i >= 0; --i) { ind += (current_index[i] % shape[i]) * mul; mul *= shape[i]; } return ind; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)