- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for add_shape (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto paddings_value = builder->create<TF::ConstOp>(filter.getLoc(), paddings); std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, filter_shape[1] + pad_w, filter_shape[2], filter_shape[3]}; SmallVector<int64_t, 4> expand_shape(pad_shape.begin(), pad_shape.end()); auto expand_result_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
%tl = "tf.TensorListFromTensor"(%arg0, %elem_shape) : (tensor<10x8x9xf32>, tensor<2xi32>) -> tensor<!tf_type.variant<tensor<8x9xf32>>> // CHECK: %[[IND_SHAPE:.*]] = "tf.Const"() <{value = dense<[5, 1]> : tensor<2xi32>}> : () -> tensor<2xi32> // CHECK: %[[IND_RESHPE:.*]] = "tf.Reshape"(%[[ARG1]], %[[IND_SHAPE]]) : (tensor<5xi32>, tensor<2xi32>) -> tensor<5x1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
return failure(); } // It is already a 1-D constant, no change. auto old_shape = shape.getShapedType().getShape(); if (old_shape.size() == 1) { return failure(); } // Verify all the leading dimensions are length one, except the last one. for (auto it = ++old_shape.rbegin(); it != old_shape.rend(); ++it) { if (*it != 1) { reshape->emitError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
Type output_type = op.getResult().getType(); const int64_t rank = operand_type.getRank(); ArrayRef<int64_t> idx_shape(rank); TensorType idx_type = operand_type.cloneWith(idx_shape, rewriter.getI32Type()); ArrayRef<int64_t> start_idx_i64 = op.getStartIndices(); ArrayRef<int64_t> limit_idx_i64 = op.getLimitIndices();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
Arg<TF_NumberTensor, [{1-D. The `values` of the first `SparseTensor`, size `[nnz]` Vector.}]>:$a_values, Arg<TF_Int64Tensor, [{1-D. The `shape` of the first `SparseTensor`, size `[ndims]` Vector.}]>:$a_shape, Arg<TF_Int64Tensor, [{2-D. The `indices` of the second `SparseTensor`, size `[nnz, ndims]` Matrix.}]>:$b_indices, Arg<TF_NumberTensor, [{1-D. The `values` of the second `SparseTensor`, size `[nnz]` Vector.}]>:$b_values,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)