Searched refs:inp_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 118 inp_shape = array_ops.shape(inp)
121 [check_ops.assert_equal(batch_size, inp_shape[0])]):
141 new_shape = array_ops.concat([output_shape, inp_shape[1:]], axis=0)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 171 const TensorShape inp_shape = ctx->InputShape(1 + i); in Compile() local
172 OP_REQUIRES(ctx, inp0_rank == inp_shape.num_elements(), in Compile()
175 inp_shape.num_elements())); in Compile()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_test.py | 638 for layer_type, inp_shape, arg_dict in _LAYERS_TO_TEST:
644 (name, layer_type, inp_shape, {k: v for k, v in args}))
|