| /external/tensorflow/tensorflow/contrib/kernel_methods/python/mappers/ |
| random_fourier_features.py | 124 input_tensor_shape = input_tensor.get_shape()
125 if len(input_tensor_shape) != 2:
128 len(input_tensor_shape))
130 features_dim = input_tensor_shape[1]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| depthtospace_op.cc | 44 const TensorShape input_tensor_shape = ctx->InputShape(0); variable
45 int input_rank = input_tensor_shape.dims();
51 input_tensor_shape.dim_sizes();
|
| spacetodepth_op.cc | 44 const TensorShape input_tensor_shape = ctx->InputShape(0); variable
45 int input_rank = input_tensor_shape.dims();
51 input_tensor_shape.dim_sizes();
|
| batchtospace_op.cc | 25 const TensorShape& input_tensor_shape,
28 const int input_rank = input_tensor_shape.dims();
30 input_tensor_shape.dim_sizes();
|
| spacetobatch_op.cc | 25 const TensorShape& input_tensor_shape,
28 const int input_rank = input_tensor_shape.dims();
30 input_tensor_shape.dim_sizes();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.h | 156 const TensorShape& input_tensor_shape) {
159 this->data_format_tf_, input_tensor_shape);
201 TensorShape input_tensor_shape = input_tensor.shape(); local
205 : memory::desc(TFShapeToMklDnnDimsInNCHW(input_tensor_shape,
210 input_tensor_shape);
333 const TensorShape& input_tensor_shape) {
337 input_tensor_shape);
342 : TFShapeToMklDnnDimsInNCHW(input_tensor_shape,
|
| mkl_avgpooling_op.cc | 654 MklPoolParameters* pool_params, TensorShape* input_tensor_shape) {
657 CHECK_NOTNULL(input_tensor_shape);
662 input_tensor_shape->AddDim(shape_vec(i));
667 original_input_dims_mkl_order, pool_params, *input_tensor_shape);
|
| mkl_maxpooling_op.cc | 726 MklPoolParameters* pool_params, TensorShape* input_tensor_shape) {
727 *input_tensor_shape = tensor_original_input.shape();
730 original_input_dims_mkl_order, pool_params, *input_tensor_shape);
[all...] |
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| periodic_resample_op.h | 103 const auto& input_tensor_shape = input_tensor.shape(); local
117 context, desired_shape[i] >= input_tensor_shape.dim_size(i),
122 i, " input tensor has size ", input_tensor_shape.dim_size(i),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_instruction_test.cc | 1275 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); local
[all...] |
