- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for xaxis (0.12 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} else { // Generalize differing axis lengths to "variable"/"unknown". for (int axis_index = 0; axis_index < combined_shape.dims(); ++axis_index) { int64_t axis_length = combined_shape.dim_size(axis_index); if (axis_length != component_shape.dim_size(axis_index)) { axis_length = -1; } TF_RETURN_IF_ERROR(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// Non-parallel tensors from _EagerConst/tf.constant are implicitly // broadcast, i.e. set as the input to each parallel operation. This // allows code like "tf.constant(1.)" or "tf.reduce_sum(..., axis=1)" // (where the value starts on the host), without allowing other implicit // copies/broadcasts. Other implicit copies may be supported eventually,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, TFE_TensorHandle* axis) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "Min", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(op, input, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(op, axis, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api_test.cc
TFE_TensorHandle* axis = TestAxisTensorHandle(ctx); TFE_Op* minOp = MinOp(ctx, input, axis); TFE_TensorHandle* retvals[1] = {nullptr}; int num_retvals = 1; TFE_Execute(minOp, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteOp(minOp); TFE_DeleteTensorHandle(input); TFE_DeleteTensorHandle(axis); ASSERT_EQ(1, num_retvals);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
// Make a placeholder operation. TF_Operation* feed = Placeholder(graph, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); // Make a constant operation with the scalar "0", for axis. TF_Operation* one = ScalarConst(0, graph, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); // Create a session for this graph. CSession csession(graph, s, use_XLA);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)