- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 28 for shape_b (0.18 sec)
-
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 20:23:51 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fold_broadcast.cc
// Gets the broadcasted output shape for tf.BatchMatMulV2Op. `shape_x` is the // shape of op's first/left-hand-side operand and `shape_y` is the shape of // op's second/right-hand-side operand. const auto get_broadcasted_shape = [&](ArrayRef<int64_t> shape_x, ArrayRef<int64_t> shape_y, SmallVectorImpl<int64_t>& result_shape) { if (shape_x.size() < 2 || shape_y.size() < 2) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// CHECK-NEXT: %[[SHAPE_0:.*]] = "tf.Const"() // CHECK: {ici_weight_distribution_mlir_bridge_marker = true} // CHECK-NEXT: %[[FULL_0:.*]] = "tf.Fill"(%[[SHAPE_0]], %[[ELEM_0]]) {ici_weight_distribution_mlir_bridge_marker = true} // CHECK: %[[ELEM_1:.*]] = "tf.Const"() // CHECK: {ici_weight_distribution_mlir_bridge_marker = true} // CHECK-NEXT: %[[SHAPE_1:.*]] = "tf.Const"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.td
(HasNoUseOf:$root__3), (HasNoUseOf:$root__4), (HasNoUseOf:$root__5)]>; class HasEqualElementSize<list<int> shape_1, list<int> shape_2> : Constraint< CPred<"quant::HasEqualElementSize($0, $1," "llvm::ArrayRef<int>({" # !interleave(shape_1, ", ") # "})," "llvm::ArrayRef<int>({" # !interleave(shape_2, ", ") # "}))">, "Checks if the given dimensions contain the same number of elements.">; def ReshapableTo1DTensor : Constraint<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 03:24:59 UTC 2024 - 8.4K bytes - Viewed (0) -
test/typeparam/shape1.go
Matthew Dempsky <******@****.***> 1646087539 -0800
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Mar 01 19:45:34 UTC 2022 - 855 bytes - Viewed (0) -
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/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)