OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:diff_dst_tensor
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc
484
const Tensor&
diff_dst_tensor
= MklGetInput(context, diff_dst_index);
local
528
auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(
diff_dst_tensor
.shape(),
553
diff_dst.SetUsrMem(diff_dst_md, &
diff_dst_tensor
);
673
const Tensor&
diff_dst_tensor
= MklGetInput(context, diff_dst_index);
local
682
diff_dst_tensor
.shape(), dnn_shape_diff_src);
687
static_cast<void*>(const_cast<T*>(
diff_dst_tensor
.flat<T>().data()));
740
const Tensor&
diff_dst_tensor
= MklGetInput(context, diff_dst_index);
local
749
diff_dst_tensor
.shape(), dnn_shape_diff_src);
754
static_cast<void*>(const_cast<T*>(
diff_dst_tensor
.flat<T>().data()));
811
const Tensor&
diff_dst_tensor
= MklGetInput(context, diff_dst_index)
local
[
all
...]
mkl_fused_batch_norm_op.cc
1054
const Tensor&
diff_dst_tensor
= MklGetInput(context, kDiffDstIndex);
variable
[
all
...]
Completed in 624 milliseconds