- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 57 for tensor_list (0.15 sec)
-
tensorflow/compiler/mlir/tfr/tests/control_flow.mlir
tfr.func @tf__my_pack(%values: !tfr.tensor_list, %n: i32 {tfr.name="N"}, %axis: i32 {tfr.name="axis"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i32 %eq = arith.cmpi eq, %n, %cst : i32 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:58:25 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
input_used_attrs.append(used.begin(), used.end()); continue; } if (auto tensor_list = arg_type.dyn_cast<TFRTensorListType>()) { if (first_tensor_list == -1) { first_tensor_list = arg.index(); } last_tensor_list = arg.index(); auto used = tensor_list.getAttrKeys(); input_used_attrs.append(used.begin(), used.end()); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/tfr_gen.py
ssa_value, _ = value self.symbol_table.insert_symbol(key, ssa_value, t) elif len(values) == 1: name, tys = values[0] if ty == TFRTypes.TENSOR_LIST: # assign single tensor_list to multiple variables for idx, key in enumerate(targets): idx_name = self._ssa_name('idx') self._emit_with_loc(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 27 15:27:03 UTC 2022 - 55.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
// Create the new operands. This is mapping the operands from the target // TF ops to the TFR function arguments. If the TFR function argument is // a tensor_list, a "tfr.build_list" op is used to concat the available // TF op operands. If the TFR function argument isn't a tensor/tensor_list, // a constant is created by using the attribute stored in the TF op or the // default value in the argument attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_types.h
}; class TFRTensorListType : public detail::TFRTypeImpl<TFRTensorListType> { public: using TFRBase::TFRBase; static constexpr StringLiteral name = "tfr.tensor_list"; static std::string getTypeName() { return "TFRTensorListType"; } }; class TFRAttrType : public Type::TypeBase<TFRAttrType, TFRType, TypeStorage> { public: using Base::Base;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 07:17:01 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
SmallVector<Value, 4> tensor_list; for (int i = res.index(); i < new_op->getNumResults(); i++) { Value new_res = new_op->getResult(i); auto casted = rewriter.create<CastOp>(loc, unconstrainted_type, new_res); tensor_list.push_back(casted.getOut()); } auto list_op = rewriter.create<BuildListOp>(loc, res_type, tensor_list); new_results.push_back(list_op.getOut());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir
// CHECK: [[tensor_list:%.*]] = "tf.TensorListReserve"([[shape:%.*]], [[reserve_size:%.*]]) { %0 = "tf.TensorListReserve"(%cst_1, %cst_2) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xf32>>> // CHECK: [[map_fn_result:%.*]] = tf_mlrt.tf_map_fn([[reserve_size]], [[tensor_list]], %arg0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:40:22 UTC 2024 - 68.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
output << " input_index=" << xla_output_description.input_index; if (xla_output_description.is_constant) output << " constant"; if (xla_output_description.is_tensor_list) output << " tensor_list"; output << '\n'; } for (const auto& resource_update : compilation_result.resource_updates) { output << "// ResourceUpdate input_index=" << resource_update.input_index
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
if (a.TypeId() == tensorflow::TypeIndex::Make<::tensorflow::TensorList>()) { TF_RETURN_IF_ERROR(ValidateVariantType<::tensorflow::TensorList>(a)); *out = ::tensorflow::TensorList(); return ::tensorflow::TensorListBinaryAdd( cc_ctx, *a.get<::tensorflow::TensorList>(), *b.get<::tensorflow::TensorList>(), out->get<::tensorflow::TensorList>(), cc_binary_add);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// Returns a `llvm::DenseMap` which maps from the index of tensorlist in the // result, to the index of the same tensorlist in the arguments. For `If` op's // branch functions, the results and arguments are not usually matched 1-1. This // will let us konw which tensorlist result maps to which tensorlist in the // arguments. Once we know this info it will help us decide the types of the // result tensorlist based on the operand's of the `If` op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)