MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MinOp

Function MinOp

tensorflow/c/eager/c_api_test_util.cc:189–205  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

187}
188
189TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input,
190 TFE_TensorHandle* axis) {
191 TF_Status* status = TF_NewStatus();
192
193 TFE_Op* op = TFE_NewOp(ctx, "Min", status);
194 CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
195 TFE_OpAddInput(op, input, status);
196 CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
197 TFE_OpAddInput(op, axis, status);
198 CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
199 TFE_OpSetAttrBool(op, "keep_dims", 1);
200 TFE_OpSetAttrType(op, "Tidx", TF_INT32);
201 TF_DeleteStatus(status);
202 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input));
203
204 return op;
205}
206
207bool GetDeviceName(TFE_Context* ctx, string* device_name,
208 const char* device_type) {

Callers 3

TESTFunction · 0.70
Execute_Min_XLA_CPUFunction · 0.70
TEST_FFunction · 0.50

Calls 9

TF_NewStatusFunction · 0.85
TFE_NewOpFunction · 0.85
TF_GetCodeFunction · 0.85
TF_MessageFunction · 0.85
TFE_OpAddInputFunction · 0.85
TFE_OpSetAttrBoolFunction · 0.85
TFE_OpSetAttrTypeFunction · 0.85
TF_DeleteStatusFunction · 0.85
TFE_TensorHandleDataTypeFunction · 0.85

Tested by

no test coverage detected