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

Method ConstantInputAsIntVector

tensorflow/compiler/tf2xla/xla_op_kernel.cc:276–281  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

274}
275
276Status XlaOpKernelContext::ConstantInputAsIntVector(int index,
277 std::vector<int64>* out) {
278 xla::Literal literal;
279 TF_RETURN_IF_ERROR(ConstantInput(index, &literal));
280 return LiteralToInt64Vector(literal, out);
281}
282
283Status XlaOpKernelContext::ConstantInputAsIntVector(absl::string_view name,
284 std::vector<int64>* out) {

Callers 15

CompileMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80
ProcessDiagIndexFunction · 0.80
CompileMethod · 0.80
GetKernelSizeMethod · 0.80
GetStrideMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80

Calls 1

LiteralToInt64VectorFunction · 0.85

Tested by

no test coverage detected