- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 74 for _input_ (0.15 sec)
-
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
RunOptions run_options; Tensor input_a = CreateInputTensor(shape, 0); Tensor input_b = CreateInputTensor(shape, shape.num_elements()); Tensor input_c = CreateInputTensor(shape, 2 * shape.num_elements()); TF_RETURN_IF_ERROR(session->Run( run_options, {std::make_pair("a", input_a), std::make_pair("b", input_b), std::make_pair("c", input_c)}, {"m"}, {}, &golden_output_tensors, nullptr));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K 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/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/cc/framework/while_gradients_test.cc
for (int i = 0; i < num_inputs; ++i) { inputs_.push_back(ops::Placeholder(scope_, dtype)); } } void CreateLoop(const ops::CondGraphBuilderFn& cond, const ops::BodyGraphBuilderFn& body, const std::vector<Output>* inputs = nullptr) { if (inputs == nullptr) inputs = &inputs_; TF_ASSERT_OK(ops::BuildWhileLoop(scope_, *inputs, cond, body, "test_loop",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 27 20:32:17 UTC 2017 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/switch_n.pbtxt
} } } node { name: "Case/branch0/_0/mul_0" op: "Mul" input: "Case/Case/input_0/_7" input: "Case/branch0/_0/mul/y" attr { key: "T" value { type: DT_FLOAT } } } node { name: "Case/branch1/_1/mul_0" op: "Mul" input: "Case/Case/input_0/_7:1" input: "Case/branch1/_1/mul/y" attr { key: "T" value { type: DT_FLOAT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 15 19:42:47 UTC 2021 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
(MHLO_CompositeOp:$old_val (variadic $a_input), ConstantStrAttr<StrAttr, "aten.avg_pool2d.default">, $attrs, $_, $_), (TFL_TransposeOp (TFL_AveragePool2DOp /*input*/ (TFL_TransposeOp $a_input, (Arith_ConstantOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/parallel_execute_to_islands.mlir
} func.return } // CHECK: [[INPUT_A:%.+]], {{%.+}} = tf_executor.island { // CHECK-NEXT: [[OP_A_OUTPUT:%.+]] = "tf.opA"([[ARG_0]]) // CHECK-NEXT: tf_executor.yield [[OP_A_OUTPUT]] : // CHECK: [[ISLAND_0_OUTPUT:%.+]], {{%.+}} = tf_executor.island { // CHECK-NEXT: [[OP_B_OUTPUT:%.+]] = "tf.opB"([[INPUT_A]]) {_parallel_execution_ids = "p0:0"} // CHECK: tf_executor.yield [[OP_B_OUTPUT]] :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 9.2K bytes - Viewed (0)