HomeSort by relevance Sort by last modified time
    Searched defs:b_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
list_kernels.h 204 const Tensor& b_tensor = b.tensors[i]; local
206 out->tensors.push_back(b_tensor);
209 if (b_tensor.dtype() == DT_INVALID) {
213 if (a_tensor.shape() != b_tensor.shape()) {
219 b_tensor.shape().DebugString(), " in position ", i);
229 a_tensor.flat<dtype>() + b_tensor.flat<dtype>(); \
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.cc 262 const Tensor* b_tensor = nullptr; variable
263 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor));
297 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4,
299 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0),
358 wcf_tensor->vec<T>(), wco_tensor->vec<T>(), b_tensor->vec<T>(),
441 const Tensor* b_tensor = nullptr; variable
442 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor));
500 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4,
502 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0),
641 wco_tensor->vec<T>(), b_tensor->vec<T>(), i_tensor->matrix<T>()
907 const Tensor* b_tensor = nullptr; variable
1081 const Tensor* b_tensor = nullptr; variable
    [all...]

Completed in 79 milliseconds