- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 45 for tensor_list (0.19 sec)
-
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir
// RUN: tf-opt "-tfl-lower-static-tensor-list=allow-tensorlist-pass-through default-to-single-batch" -split-input-file %s | FileCheck %s // ----- // CHECK-LABEL: tensorlistConst func.func @tensorlistConst(%arg0 : tensor<1xi32>) -> tensor<2x3xi32> { // CHECK-DAG: %[[ELEMENT0:.*]] = "tf.Const"() <{value = dense<[0, 1, 2]> : tensor<3xi32>}> : () -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 39.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
} for (const XlaOutputDescription& descr : compilation_result->outputs) { if (descr.type == DT_VARIANT) { return errors::Unimplemented( "Support for TensorList crossing the XLA/TF boundary " "is not implemented"); } } std::vector<TensorShape> output_tensor_shapes; output_tensor_shapes.reserve(ctx->num_outputs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
const int64_t id = target_tensor_ids[i]; if (output_gradients.empty() || output_gradients[i] == nullptr) { auto tensor_it = tensor_tape.find(id); if (tensor_it != tensor_tape.end() && tensor_it->second != -1) { auto op_it = op_tape.find(tensor_it->second); if (op_it == op_tape.end()) { return errors::Internal( "Internal state of the gradient tape is invalid: "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
func_output_shape, mlir::cast<RankedTensorType>(input_.getType()).getElementType()); auto tensor_cast = builder_.create<mlir::tensor::CastOp>( fused_func_op_.getLoc(), func_result_type, lstm_.getResult()); builder_.create<mlir::func::ReturnOp>(fused_func_op_.getLoc(), tensor_cast.getResult()); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
mlir::TF::RecvTPUEmbeddingActivationsOp::getOperationName(), context)); supported_ops->insert(OperationName( mlir::TF::SendTPUEmbeddingGradientsOp::getOperationName(), context)); } // Stack, TensorList and TensorArray ops are rewritten during the second phase // of the bridge (compilation of TPUCompile op). They would not match any // legalization/canonicalization pattern and have to be manually added to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// Materialize the MlirPassthroughOp by replacing it with the MLIR module // attached as an attribute. std::unique_ptr<OperationPass<func::FuncOp>> CreateMaterializePassthroughOpPass(); // Replicates the TensorList init op by undoing some CSE needed for correct // shape assignment in shape_inference. std::unique_ptr<OperationPass<func::FuncOp>> CreateReplicateTensorListInitOpsPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
*tensor_proto = TensorProtoHelpers::GenerateTensorProto(); CreateAndCallKernelWithAttr(my_create_func, "TestKernelAttrTensor", v); } TEST_F(TestKernelAttr, TensorList) { struct TensorProtoHelpers { public: static ::tensorflow::TensorProto GenerateTensorProto1() { ::tensorflow::TensorProto tensor_proto; tensor_proto.mutable_tensor_shape()->add_dim()->set_size(2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
"return getElementTypeOrSelf(element_type());">; let extraClassDeclaration = [{ // Returns type of the TensorList element produced by this op. TensorType element_type() { return handle_dtype().getSubtypes()[0]; } // Returns data type of the result handle. Returned type contains type of // the TensorList element as a subtype. VariantType handle_dtype() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK: %[[CAST:.*]] = "tf.Cast"(%[[ADD]]) {{.*}} : (tensor<10xf32>) -> tensor<*xf32> // CHECK: %[[TENSOR_CAST:.*]] = tensor.cast %[[CAST]] : tensor<*xf32> to tensor<?xf32> // CHECK: return %[[TENSOR_CAST]] : tensor<?xf32> %0 = "tf.Mul"(%arg0, %arg1) : (tensor<1xf32>, tensor<10xf32>) -> tensor<*xf32> %1 = "tf.Add"(%0, %0) : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
LogicalResult matchAndRewrite(Operation *op, PatternRewriter &rewriter) const override { auto addn_op = cast<AddNOp>(op); // TODO(hinsu): Support variant with TensorList type. tf.AddV2 doesn't // support variant type so variant types require special handling. if (mlir::isa<VariantType>(getElementTypeOrSelf(addn_op.getType()))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0)