Home
last modified time | relevance | path

Searched refs:IsTensorListInput (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Didentity_op.cc29 if (IsTensorListInput(ctx, i)) { in Compile()
Dwhile_op.cc100 if (IsTensorListInput(ctx, i)) { in MakeXlaCompilerArgumentsFromInputs()
413 if (arg.initialized || !IsTensorListInput(ctx, i)) { in Compile()
502 } else if (IsTensorListInput(ctx, input_num)) { in Compile()
582 if (IsTensorListInput(ctx, i)) { in Compile()
Dtensor_list_utils.h26 bool IsTensorListInput(XlaOpKernelContext* ctx, int index);
Dshape_op.cc340 if (IsTensorListInput(ctx, 0)) { in Compile()
Dtensor_list_ops.cc638 bool element_is_tensor_list = IsTensorListInput(ctx, 1); in Compile()
Dtensor_list_utils.cc111 bool IsTensorListInput(XlaOpKernelContext* ctx, int index) { in IsTensorListInput() function