/external/tensorflow/tensorflow/core/kernels/ |
H A D | matching_files_op.cc | 52 Tensor* output_t = nullptr; variable 55 &output_t)); 56 auto output = output_t->vec<string>();
|
H A D | edit_distance_op.cc | 154 auto output_t = output->flat<float>(); variable 155 output_t.setZero(); 183 output_t(loc) = 185 if (normalize_) output_t(loc) /= truth_seq.size(); 192 output_t(loc) = hypothesis_seq.size(); 193 if (normalize_ && output_t(loc) != 0.0f) { 194 output_t(loc) = std::numeric_limits<float>::infinity(); 200 output_t(loc) = (normalize_) ? 1.0 : truth_seq.size(); 210 output_t(loc) = hypothesis_seq.size(); 211 if (normalize_ && output_t(lo [all...] |
H A D | bincount_op.cc | 106 Tensor* output_t; variable 108 ctx->allocate_output(0, TensorShape({size}), &output_t)); 109 auto output = output_t->flat<T>();
|
H A D | population_count_op.cc | 47 Tensor* output_t; variable 48 OP_REQUIRES_OK(c, c->allocate_output(0, input_t.shape(), &output_t)); 51 auto output = output_t->flat<uint8>();
|
H A D | one_hot_op.cc | 113 auto output_t = variable 118 off_value_t, &output_t);
|
H A D | compare_and_bitpack_op.cc | 63 Tensor* output_t; variable 64 OP_REQUIRES_OK(c, c->allocate_output(0, output_shape, &output_t)); 68 auto output = output_t->flat_inner_dims<uint8>();
|
H A D | xsmm_conv2d_test.cc | 136 LIBXSMM_VLA_DECL(4, float, output_t, output + (pad_w_out * ofwp + pad_h_out), 153 LIBXSMM_VLA_ACCESS(4, output_t, img, ofm, oj, oi, nOfm, ofhp,
|
/external/harfbuzz_ng/util/ |
H A D | shape-consumer.hh | 34 template <typename output_t> 94 output_t output;
|
/external/tensorflow/tensorflow/contrib/image/kernels/ |
H A D | image_ops.cc | 85 Tensor* output_t; variable 86 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, images_t.shape(), &output_t)); 87 auto output = output_t->tensor<T, 4>();
|
H A D | segmentation_ops.cc | 52 Tensor* output_t; variable 53 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, images_t.shape(), &output_t)); 65 ctx, output_t->flat<OutputType>(), images, forest, rank);
|
/external/tensorflow/tensorflow/contrib/graph_editor/tests/ |
H A D | match.py | 113 for output_t, output_op_matches in zip(op.outputs, 117 if len(output_t.consumers()) != len(output_op_matches): 119 for consumer_op, consumer_op_match in zip(output_t.consumers(),
|
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
H A D | quantile_ops.cc | 115 Tensor* output_t = nullptr; local 119 TensorShape({num_values, 2}), &output_t)); 121 auto output = output_t->matrix<int32>(); 198 Tensor* output_t = nullptr; local 201 index, {static_cast<int64>(boundaries.size())}, &output_t)); 202 auto* quantiles_flat = output_t->flat<float>().data(); 391 Tensor* output_t = nullptr; 392 OP_REQUIRES_OK(context, output_list->allocate(index, {}, &output_t)); 393 summary_proto->SerializeToString(&output_t->scalar<string>()()); 665 Tensor* output_t variable [all...] |
/external/tensorflow/tensorflow/contrib/graph_editor/ |
H A D | edit.py | 121 for output_id, output_t in enumerate(sgv.outputs) 122 if output_t.consumers()])
|
H A D | transform.py | 517 for output_t in sgv.outputs: 518 if output_t not in info.transformed_ts: 520 output_t_ = info.transformed_ts[output_t]
|
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
H A D | model_ops.cc | 139 Tensor* output_t = nullptr; variable 141 context->allocate_output(0, TensorShape(), &output_t)); 142 output_t->scalar<int32>()() =
|
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | map_and_batch_dataset_op.cc | 229 auto output_t = output->flat_outer_dims<type>(); \ 232 output_t.template chip<0>(i) = value_t.template chip<0>(i); \
|
/external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
H A D | cudnn_rnn_ops.cc | 585 Tensor* output_t = nullptr; variable 586 OP_REQUIRES_OK(context, context->allocate_output(0, {1}, &output_t)); 587 *output_t->template flat<Index>().data() = params_size;
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
H A D | layers_test.py | 2848 output_t = _layers.layer_norm( 2860 outputs, beta, gamma = sess.run((output_t, beta_var, gamma_var))
|