- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 56 for input_ (0.11 sec)
-
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_ops_test.py
} self._assertOpAndComposite([input_, filter_, bias], tf.function(gen_mnist_ops.new_conv2d), ops_defs._composite_conv_add_relu, kwargs) def test_new_conv2d_relu6(self): input_ = tf.random.uniform([1, 4, 4, 1]) filter_ = tf.random.uniform([2, 2, 1, 8]) bias = tf.zeros([8]) kwargs = { 'input_': input_, 'filter_': filter_,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
] @Composite( 'NewFullyConnected', inputs=['input_: T', 'filter_: T', 'bias: T'], attrs=['act: {"", "RELU", "RELU6", "TANH"} = ""'], derived_attrs=['T: {float, int8}'], outputs=['o: T']) def _composite_fully_connected(input_, filter_, bias, act): res = tf.raw_ops.MatMul( a=input_, b=filter_, transpose_a=False, transpose_b=True) res = tf.raw_ops.Add(x=res, y=bias)
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/tfr/examples/pad/pad_ops_test.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/pad/ops_defs.py
num_split=2) input_ = tf.raw_ops.Concat( concat_dim=i, values=[left_padding, input_, right_padding]) return input_ @tf.RegisterGradient('NewMirrorPad') def _mirror_pad_grad(op, grad): mode = op.get_attr('mode') return [gen_array_ops.mirror_pad_grad(grad, op.inputs[1], mode=mode), None] @Composite( 'NewMirrorPadGrad', inputs=['input_: T', 'paddings: Tpaddings'],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 01 05:00:29 UTC 2021 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/README.md
```python import tensorflow as tf @Composite( 'FusedFullyConnected', inputs=['input_: T', 'filter_: T', 'bias: T'], attrs=['act: {"", "RELU", "RELU6", "TANH"} = ""'], derived_attrs=['T: {float, int8}'], outputs=['o: T']) def _composite_fully_connected(input_, filter_, bias, act): res = tf.raw_ops.MatMul( a=input_, b=filter_, transpose_a=False, transpose_b=True) res = tf.raw_ops.Add(x=res, y=bias)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 29 18:32:13 UTC 2022 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/op_reg_gen.py
.format(op_name, expected_args, all_func_args)) cxx_reg_code = ['\nREGISTER_OP("{}")'.format(op_name)] for input_ in inputs: cxx_reg_code.append('.Input("{}")'.format(input_)) for attr in attrs: py_str = attr.replace('"', "'") cxx_reg_code.append('.Attr("{}")'.format(py_str)) for attr in all_dec_args.get('derived_attrs', []):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.h
func::FuncOp fused_func_op_; Value input_; Value weight_; Value bias_; Value projection_; bool couple_input_forget_gates_; // internal state Value weight_transposed_; Value projection_transposed_; RankedTensorType weight_type_; RankedTensorType projection_type_; int num_gates_; int n_cell_; int n_output_; int n_input_; int num_cols_weight_transposed_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_device_ops.mlir
%10 = "tf.opK"() : () -> tensor<*xi16> %11 = "tf.opL"() : () -> tensor<*xi64> tf_device.replicate([%0, %1, %2] as %input0: tensor<*xi1>, %9 as %input1: tensor<*xi8>, %10 as %input2: tensor<*xi16>, [%3, %4, %5] as %input3: tensor<*xi32>, [%6, %7, %8] as %input4: tensor<*xf32>, %11 as %input5: tensor<*xi64>) {n = 3 : i32} { tf_device.return } func.return // CHECK: %[[OP_A:[a-z0-9]*]] = "tf.opA"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/e2e/simple-graph.mlir
// RUN: tac-translate -input-mlir -output-mlir -device-specs=GPU %s -o - 2>&1 | FileCheck %s module { func.func @main(%arg0: tensor<1xf32>, %arg1: tensor<1xf32>, %arg2: tensor<1xf32>, %arg3: tensor<1xf32>) -> tensor<2x1xf32> attributes {tf.entry_function = {inputs = "input0,input1,input2,input3", outputs = "output"}} { %0 = "tfl.add"(%arg0, %arg1) {fused_activation_function = "RELU6"} : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 1.6K bytes - Viewed (0) -
platforms/software/dependency-management/src/test/groovy/org/gradle/internal/rules/RuleSourceBackedRuleActionTest.groovy
void theRule(List subject, String input1, Integer input2, Set input3) { subject.add(input1) subject.add(input2) subject.addAll(input3) } } static class ArrayListRuleSource { @Mutate void theRule(ArrayList subject, String input1, Integer input2, Set input3) { subject.add(input1) subject.add(input2) subject.addAll(input3)
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue Oct 10 21:10:11 UTC 2023 - 6.4K bytes - Viewed (0)