- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 236 for tensor_a (0.36 sec)
-
tensorflow/compiler/jit/xla_tensor.h
// Create a DeviceMemoryBase from a Tensor. The Tensor can be an XlaTensor, in // which case the returned value is shaped_buffer()->root_buffer(), or a // normal Tensor in which case the returned value is // {tensor.tensor_data().data(), tensor.tensor_data().size}. static se::DeviceMemoryBase DeviceMemoryFromTensor(const Tensor& tensor); // Assign the internal ShapedBuffer to new memory for the given dtype and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/tensor-list.pbtxt
# CHECK: tf.TensorListReserve{{.*}}(tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*xf32>>> # Nested variant type. # CHECK: tf.TensorListReserve{{.*}}(tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<*x!tf_type.variant>>> # CHECK: tf.TensorListSetItem{{.*}}(tensor<!tf_type.variant<tensor<*xf32>>>, tensor<i32>, tensor<2x2xf32>) -> tensor<*x!tf_type.variant>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 29 04:41:05 UTC 2021 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema.fbs
// model. table SubGraph { // A list of all tensors used in this subgraph. tensors:[Tensor]; // Indices of the tensors that are inputs into this subgraph. Note this is // the list of non-static tensors that feed into the subgraph for inference. inputs:[int]; // Indices of the tensors that are outputs out of this subgraph. Note this is // the list of output tensors that are considered the product of the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
include "mlir/Dialect/Func/IR/FuncOps.td" include "mlir/Dialect/Tensor/IR/TensorOps.td" include "stablehlo/dialect/ChloOps.td" include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" include "mhlo/IR/hlo_ops.td" def SignedIntTensor : TensorOf<[I1, I8, I16, I32, I64]>; def UnsignedIntTensor : TensorOf<[UI8, UI16, UI32, UI64]>; // IEEE compliant floating point tensors. def IEEEFloatTensor : TensorOf<[F16, F32, F64]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.td
); let results = (outs TensorOf<[TF_Float32]>:$output, TensorOf<[TF_Float32]>:$min, TensorOf<[TF_Float32]>:$max, TensorOf<[TF_Int64]>:$histogram ); } def TF_CalibrationStatisticsSaverOp : TF_Op<"CalibrationStatisticsSaver", []> { let summary = "Aggregates and saves calibration statistics."; let arguments = (ins Variadic<TensorOf<[TF_Float32, TF_Int64]>>:$inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
return %0 : tensor<1x1x1xf32> } //CHECK:func.func private @slice(%arg0: tensor<160x20x1xf32>) -> tensor<1x1x1xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
Type qtype = attr.getValue(); auto tensor_or = BuildTensorFromType( qtype, name_mapper_.GetUniqueName(intermediate).str()); if (!tensor_or.has_value()) { continue; } else { intermediates.push_back(tensors.size()); tensors.push_back(tensor_or.value()); } } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir
%0 = "tf.TensorListFromTensor"(%arg0, %arg1) : (tensor<*xf32>, tensor<1xi32>) -> tensor<!tf_type.variant<tensor<*xf32>>> %1 = "tf.TensorListGetItem"(%0, %arg2, %arg1) : (tensor<!tf_type.variant<tensor<*xf32>>>, tensor<i32>, tensor<1xi32>) -> tensor<*xf32> %2 = "tf.TensorListStack"(%0, %arg1) : (tensor<!tf_type.variant<tensor<*xf32>>>, tensor<1xi32>) -> tensor<*xf32> func.return %1, %2 : tensor<*xf32>, tensor<*xf32>
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/mlir/tfr/ir/tfr_ops.td
def TFR_CastOp : TFR_Op<"cast", [Pure]> { let description = [{ The `cast` operation converts the operand with built-in tensor type to tfr.tensor type, or vice versa. Example: ```mlir %1 = tfr.cast(%0) : tensor<f32> -> !tfr.tensor %3 = tfr.cast(%1) : !tfr.tensor -> tensor<f32> ``` }]; let arguments = (ins TFR_singleTensorType:$arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td
Note: Additional result corresponds to the control output. }]; let arguments = (ins TensorOf<[I1]>:$input, Variadic<TfeControlType>:$controlInputs ); let results = (outs TensorOf<[I1]>:$output, TfeControlType:$control ); let hasVerifier = 0; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 22K bytes - Viewed (0)