Searched refs:stamp_token_t (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
D | stats_accumulator_ops.cc | 361 const Tensor* stamp_token_t; in Compute() local 362 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 369 result->set_stamp(stamp_token_t->scalar<int64>()()); in Compute() 388 const Tensor* stamp_token_t; in Compute() local 389 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 402 result->set_stamp(stamp_token_t->scalar<int64>()()); in Compute() 437 const Tensor* stamp_token_t; in Compute() local 438 OP_REQUIRES_OK(context, context->input(kStampTokenName, &stamp_token_t)); in Compute() 439 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 502 const Tensor* stamp_token_t; in Compute() local [all …]
|
D | model_ops.cc | 39 const Tensor* stamp_token_t; in Compute() local 40 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 41 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 122 const Tensor* stamp_token_t; in Compute() local 123 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 124 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 157 const Tensor* stamp_token_t; in Compute() local 158 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 159 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute()
|
D | quantile_ops.cc | 244 const Tensor* stamp_token_t; in Compute() local 245 OP_REQUIRES_OK(context, context->input(kStampTokenName, &stamp_token_t)); in Compute() 253 stamp_token_t->scalar<int64>()()); in Compute() 286 const Tensor* stamp_token_t; in Compute() local 287 OP_REQUIRES_OK(context, context->input(kStampTokenName, &stamp_token_t)); in Compute() 288 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 507 Tensor* stamp_token_t = nullptr; in Compute() local 510 &stamp_token_t)); in Compute() 511 stamp_token_t->scalar<int64>()() = stamp_token; in Compute() 575 const Tensor* stamp_token_t = nullptr; in Compute() local [all …]
|
D | training_ops.cc | 203 const Tensor* stamp_token_t; in Compute() local 204 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 205 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 329 const Tensor* stamp_token_t; in Compute() local 330 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 331 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 913 const Tensor* stamp_token_t; in Compute() local 914 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 915 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute()
|
/external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
D | resource_ops.cc | 42 const Tensor* stamp_token_t; in Compute() local 43 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 44 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() 179 const Tensor* stamp_token_t; in Compute() local 180 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute() 181 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute()
|