- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for Input (0.16 sec)
-
tensorflow/c/eager/c_api_test_util.h
// Return a shape op fetching the shape of `a`. TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a); // Return an allreduce op adding up input tensor `in` from `group_size` workers. TFE_Op* AllReduceOp(TFE_Context* ctx, TFE_TensorHandle* in, int group_size); // Return a SendOp op `op_name` with send input tensor `in` and attributes // `send_device`, `recv_device`, and `send_device_incarnation` set.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/c_test.c
// A compute function. This will never actually get called in this test, it's // just nice to know that it compiles. void compute(void* kernel, TF_OpKernelContext* ctx) { TF_Tensor* input; TF_Status* s = TF_NewStatus(); TF_GetInput(ctx, 0, &input, s); TF_DeleteTensor(input); TF_DeleteStatus(s); } // Exercises tensorflow's C API. int main(int argc, char** argv) { TF_InitMain(argv[0], &argc, &argv);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Apr 24 20:50:35 GMT 2024 - 2.8K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.h
TF_ExecutionContextGetTFEContext(wrap(ctx), status.get()); TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get())); TFE_TensorHandle* input_eager = TestTensorHandleWithDims<T, datatype>(eager_ctx, data, dims, num_dims); *tensor = unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get())); return absl::OkStatus(); } // Return a scalar tensor handle with given value.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0)