Home
last modified time | relevance | path

Searched refs:GetVariableTypeAndShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h197 Status GetVariableTypeAndShape(int index, DataType* type,
Dxla_op_kernel.cc420 Status XlaOpKernelContext::GetVariableTypeAndShape(int index, DataType* type, in GetVariableTypeAndShape() function in tensorflow::XlaOpKernelContext
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dvariable_ops.cc52 ctx->GetVariableTypeAndShape(0, &variable_dtype, &shape)); in Compile()