Searched refs:body_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 180 TfLiteTensor* body_input = in Prepare() local
184 TF_LITE_ENSURE_TYPES_EQ(context, body_input->type, body_output->type); in Prepare()
187 if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) { in Prepare()
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_architectures.py | 70 body_input = keras.Input(shape=(None,), name='body')
73 x = keras.layers.Embedding(10, 4)(body_input)
81 inputs=[body_input, tags_input], outputs=[pred_1, pred_2])
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 333 auto body_input = GetTupleElement(input_tuple, 2); in InvertDiagonalBlocks() local
338 DynamicSlice(body_input, {zero, j, zero}, in InvertDiagonalBlocks()
355 Tuple(bodyb.get(), {next_i, body_out, body_input}); in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2836 TypeRange body_input, in VerifyWhileTypes() argument
2845 {body_input, "body input"}, in VerifyWhileTypes()
|